torch.nn.Linear.md

[ 参数不一致 ]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 的作用一致。

转写示例

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)