[ 仅参数名不一致 ]torch.argmax¶

torch.argmax ¶

          torch.argmax(input,
             dim=None,
             keepdim=False)

         

          paddle.argmax(x,
              axis=None,
              keepdim=False,
              dtype='int64',
              name=None)

         

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

PyTorch	PaddlePaddle	备注
input	x	输入的多维 Tensor ，仅参数名不一致。
dim	axis	指定进行运算的轴，仅参数名不一致。
keepdim	keepdim	是否在输出 Tensor 中保留减小的维度。
-	dtype	输出 Tensor 的数据类型， PyTorch 无此参数， Paddle 保持默认即可。