[ 参数不一致 ]torch.nn.Linear¶

torch.nn.Linear¶

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

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。 | | - | weight_attr | 指定权重参数的属性，PyTorch 无此参数，Paddle 保持默认即可。 | | - | bias_attr | 指定偏置参数的属性, 当bias_attr设置为 bool 类型时与 PyTorch 的作用一致。 |

转写示例¶

# 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)