Softmax 差异对比

torch.nn.Softmax

torch.nn.Softmax(
   dim=None
)

更多请查看 torch.nn.Softmax.

megengine.module.Softmax

megengine.module.Softmax(
   axis=None
)

更多请查看 megengine.module.Softmax.

参数差异

两者功能一致且参数用法一致，仅参数名不一致，dim 与 axis 都是对输入 Tensor 进行运算的轴，仅参数名不一致。