[ 返回参数类型不一致 ]torch.Tensor.max

该 api 有两组参数列表重载,因此有两组差异分析。


torch.Tensor.max

torch.Tensor.max(dim=None, keepdim=False)

paddle.Tensor.max

paddle.Tensor.max(axis=None, keepdim=False, name=None)

其中 PyTorch 与 Paddle 指定 dim 后返回值不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
dim axis 求最大值运算的维度, 仅参数名不一致。
keepdim keepdim 是否在输出 Tensor 中保留减小的维度。
返回值 返回值 表示返回结果,当指定 dim 后,PyTorch 会返回比较结果和元素索引, Paddle 不会返回元素索引,需要转写。

转写示例

指定 dim 后的返回值

# PyTorch 写法
result = x.max(dim=1)

# Paddle 写法
result = x.max(dim=1), x.argmax(dim=1)

torch.Tensor.max

torch.Tensor.max(other)

paddle.Tensor.maximum

paddle.Tensor.maximum(y)

两者功能一致且参数用法一致,仅参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
other y 输⼊ Tensor ,仅参数名不一致。