[ 仅 paddle 参数更多 ]torch.Tensor.softmax

torch.Tensor.softmax

torch.Tensor.softmax(dim)

paddle.nn.functional.softmax

paddle.nn.functional.softmax(x, axis=-1, dtype=None, name=None)

功能一致,torch 是类成员方式,paddle 是 function 调用,具体差异如下:

参数差异

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | dim | axis | 计算 softmax 的轴,仅参数名不一致。 |

转写示例

dim: 计算 softmax 的轴

# torch 写法
x.softmax(dim=1)

# paddle 写法
paddle.nn.functional.softmax(x, axis=1)