torch.Tensor.softmax.md

[ 仅 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)