[ 输入参数用法不一致 ]torch.nn.Linear

torch.nn.Linear

torch.nn.Linear(in_features,
                out_features,
                bias=True,
                device=None,
                dtype=None)

paddle.nn.Linear

paddle.nn.Linear(in_features,
                 out_features,
                 weight_attr=None,
                 bias_attr=None,
                 name=None)

其中 PyTorch 的 bias 与 Paddle 的 bias_attr 用法不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
in_features in_features 表示线性变换层输入单元的数目。
out_features out_features 表示线性变换层输出单元的数目。
bias - 是否在输出中添加可学习的 bias。
device - 指定 Tensor 的设备，Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。
dtype - Tensor 的所需数据类型，Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。
- weight_attr 指定权重参数的属性，PyTorch 无此参数，Paddle 保持默认即可。
- bias_attr 指定偏置参数的属性, 当bias_attr设置为 bool 类型时与 PyTorch 的作用一致。

转写示例

bias: 是否在输出中添加可学习的 bias

# PyTorch 写法
torch.nn.Linear(2, 4, bias=True)

# Paddle 写法
paddle.nn.Linear(2, 4)

# PyTorch 写法
torch.nn.Linear(2, 4, bias=False)

# Paddle 写法
paddle.nn.Linear(2, 4, bias_attr=False)