[ 输入参数用法不一致 ]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)