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

torch.nn.functional.linear

torch.nn.functional.linear(input,
                           weight,
                           bias=None)

paddle.nn.functional.linear

paddle.nn.functional.linear(x,
                            weight,
                            bias=None,
                            name=None)

两者功能一致，weight 参数的 shape 不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
input x 表示输入的 Tensor ，仅参数名不一致。
weight weight 表示权重 Tensor， PyTorch weight 参数的 shape 为 (out_features,in_features)，Paddle weight 参数的 shape 为 (in_features, out_features)，两者互为转置，需要转写。
bias bias 表示偏重 Tensor 。

转写示例

weight

# PyTorch 写法
result = F.linear(x, weight=weight)

# Paddle 写法
result = F.linear(x, weight=weight.T)