@@ -18,15 +18,15 @@ fused_feedforward
1818
1919 参数
2020:::::::::
21- - **x ** (Tensor) - 输入Tensor,数据类型支持float16, float32 和float64, 输入的形状是` [batch_size, sequence_length, d_model]`。
22- - **linear1_weight ** (Tensor) - 第一个linear算子的权重数据,数据类型与`x` 一样,形状是` [d_model, dim_feedforward]`。
23- - **linear2_weight ** (Tensor) - 第二个linear算子的权重数据,数据类型与`x` 一样,形状是` [dim_feedforward, d_model]`。
24- - **linear1_bias ** (Tensor, 可选) - 第一个linear算子的偏置数据,数据类型与`x` 一样,形状是` [dim_feedforward]`。默认值为None。
25- - **linear2_bias ** (Tensor, 可选) - 第二个linear算子的偏置数据,数据类型与`x` 一样,形状是` [d_model]`。默认值为None。
26- - **ln1_scale ** (Tensor, 可选) - 第一个layer_norm算子的权重数据,数据类型可以是float32或者float64,形状和`x` 一样。默认值为None。
27- - **ln1_bias ** (Tensor, 可选) - 第一个layer_norm算子的偏置数据,数据类型和` ln1_scale`一样, 形状是` [d_model]`。默认值为None。
28- - **ln2_scale ** (Tensor, 可选) - 第二个layer_norm算子的权重数据,数据类型可以是float32或者float64,形状和`x` 一样。默认值为None。
29- - **ln2_bias ** (Tensor, 可选) - 第二个layer_norm算子的偏置数据,数据类型和` ln2_scale`一样, 形状是`[d \_ model]` 。默认值为None。
21+ - **x ** (Tensor) - 输入Tensor,数据类型支持float16, float32 和float64, 输入的形状是 `` [batch_size, sequence_length, d_model] `` 。
22+ - **linear1_weight ** (Tensor) - 第一个linear算子的权重数据,数据类型与 `` x `` 一样,形状是 `` [d_model, dim_feedforward] `` 。
23+ - **linear2_weight ** (Tensor) - 第二个linear算子的权重数据,数据类型与 `` x `` 一样,形状是 `` [dim_feedforward, d_model] `` 。
24+ - **linear1_bias ** (Tensor, 可选) - 第一个linear算子的偏置数据,数据类型与 `` x `` 一样,形状是 `` [dim_feedforward] `` 。默认值为None。
25+ - **linear2_bias ** (Tensor, 可选) - 第二个linear算子的偏置数据,数据类型与 `` x `` 一样,形状是 `` [d_model] `` 。默认值为None。
26+ - **ln1_scale ** (Tensor, 可选) - 第一个layer_norm算子的权重数据,数据类型可以是float32或者float64,形状和 `` x `` 一样。默认值为None。
27+ - **ln1_bias ** (Tensor, 可选) - 第一个layer_norm算子的偏置数据,数据类型和 `` ln1_scale `` 一样, 形状是 `` [d_model] `` 。默认值为None。
28+ - **ln2_scale ** (Tensor, 可选) - 第二个layer_norm算子的权重数据,数据类型可以是float32或者float64,形状和 `` x `` 一样。默认值为None。
29+ - **ln2_bias ** (Tensor, 可选) - 第二个layer_norm算子的偏置数据,数据类型和 `` ln2_scale `` 一样, 形状是 `` [d_model] `` 。默认值为None。
3030 - **dropout1_rate ** (float, 可选) - 第一个dropout算子置零的概率。默认是0.5。
3131 - **dropout2_rate ** (float, 可选) - 第二个dropout算子置零的概率。默认是0.5。
3232 - **activation ** (string, 可选) - 激活函数。默认值是relu。
@@ -37,7 +37,7 @@ fused_feedforward
3737
3838返回
3939:::::::::
40- - Tensor, 输出Tensor,数据类型与`x` 一样。
40+ - Tensor, 输出Tensor,数据类型与 `` x `` 一样。
4141
4242代码示例
4343::::::::::
0 commit comments