Linear

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

    ,计算公式为:

    Linear - 图1

    其中,

    Linear - 图2

    Linear层只接受一个Tensor作为输入,形状为

    ,其中

    Linear - 图3

    表示可以为任意个额外的维度。 该层可以计算输入Tensor与权重矩阵

    的乘积,然后生成形状为 [batch_size,∗,out_features][batch_size,∗,out_features] 的输出Tensor。 如果 bias_attrbias_attr 不是False,则将创建一个偏置参数并将其添加到输出中。

    • in_features (int) – 线性变换层输入单元的数目。

    • weight_attr (ParamAttr, 可选) – 指定权重参数的属性。默认值为None,表示使用默认的权重参数属性,将权重参数初始化为0。具体用法请参见 。

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

    本层的可学习参数,类型为 。

    本层的可学习偏置,类型为 。

    • 输入:形状为 [batch_size,∗,in_features][batch_size,∗,in_features] 的多维Tensor。