fused_linear¶

paddle.incubate.nn.functional. fused_linear ( x, weight, bias=None, trans_x=False, transpose_weight=False, name=None ) [源代码] ¶

全连接线性变换算子。该方法要求 CUDA 版本大于等于 11.6 。

参数¶

x (Tensor) – 需要进行乘法运算的输入 Tensor。

weight (Tensor) – 需要进行乘法运算的权重 Tensor，它的阶数必须为 2。

bias (Tensor, 可选) – 输入的偏置 Tensor。如果为 None ，则不执行偏置加法。否则，将偏置加到矩阵乘法的结果上。默认值为 None。

transpose_weight (bool, 可选) - 是否在乘法之前转置权重。默认值：False。

name (str, 可选) - 如需详细信息，请参阅 Name 。一般无需设置，默认值为 None。

返回¶

Tensor，变换之后的 Tensor。

代码示例¶

          >>> import paddle
>>> from paddle.incubate.nn.functional import fused_linear

>>> paddle.set_device('gpu')
>>> x = paddle.randn([3, 4])
>>> weight = paddle.randn([4, 5])
>>> bias = paddle.randn([5])
>>> out = fused_linear(x, weight, bias)
>>> print(out.shape)
[3, 5]

         