\u200E

[ 仅 API 调用方式不一致 ]torch.nn.functional.softmax

torch.nn.functional.softmax 

          torch.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None)

         

paddle.compat.nn.functional.softmax 

          paddle.compat.nn.functional.softmax(input, dim=None, _stacklevel=3, dtype=None, out=None)

         

两者功能一致，但调用方式不一致，具体如下：

转写示例

          # PyTorch 写法
result = torch.nn.functional.softmax(x, -1)

# Paddle 写法
result = paddle.compat.nn.functional.softmax(x, -1)