[ torch 参数更多 ]torch.amax¶

torch.amax ¶

          torch.amax(input, dim, keepdim=False, *, out=None)

         

          paddle.amax(x, axis=None, keepdim=False, name=None)

         

PyTorch 相比 Paddle 支持更多其他参数，具体如下：

           # PyTorch 写法
torch.amax(a, dim=0，out=y)

# Paddle 写法
paddle.assign(paddle.amax(a, dim=0), y)