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