\u200E

[仅 paddle 参数更多]torch.nn.Softmax2d¶

torch.nn.Softmax2d ¶

          torch.nn.Softmax2d()

         

paddle.nn.Softmax ¶

          paddle.nn.Softmax(axis=-1)

         

其中 Paddle 并没有 torch.nn.Softmax2d 此 api ，可通过 paddle.nn.Softmax 设置参数 axis 为 -3 实现同样的效果：

参数映射¶

PyTorch	PaddlePaddle	备注
-	axis	指定对输入 Tensor 进行运算的轴。

转写示例¶

          # PyTorch 写法
cri = torch.nn.Softmax2d()
cri(input)

# Paddle 写法
cri = paddle.nn.Softmax(axis=-3)
cri(input)

         