linear

    线性变换OP 。对于每个输入Tensor

    ,计算公式为:

    linear - 图1

    其中,

    linear - 图2

    分别为权重和偏置。

    如果权重

    是一个形状为

    linear - 图3

    ,其中 ∗∗ 表示可以为任意个额外的维度。 linear接口可以计算输入Tensor与权重矩阵 WW 的乘积,生成形状为 [batch_size,∗,out_features][batch_size,∗,out_features] 的输出Tensor。 如果偏置 biasbias 不是None,它必须是一个形状为 [out_features][out_features] 的1-D Tensor,且将会被其加到输出中。

    • weight (Tensor) – 权重Tensor。它的数据类型可以为float16,float32或float64。

    • name (str,可选) – 具体用法请参见 ,一般无需设置,默认值为None。