[ 仅 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 的轴，仅参数名不一致。 |

转写示例¶

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

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