linear
线性变换OP 。对于每个输入Tensor
,计算公式为:
其中,
分别为权重和偏置。
如果权重
是一个形状为
,其中 ∗∗ 表示可以为任意个额外的维度。 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,可选) – 具体用法请参见 Name ,一般无需设置,默认值为None。