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

输入一个 Tensor 对应 paddle.max,输入两个 Tensor 对应 paddle.maximum,因此有两组差异分析,分别如下:


torch.max

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

paddle.max

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

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

参数映射

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

转写示例

out:指定输出

# 对指定维度上的 Tensor 元素求最大值运算

# PyTorch 写法
torch.max(a, out=y)
# torch 在输入 dim 时,返回 (values, indices),返回参数类型不一致

# Paddle 写法
paddle.assign(paddle.max(a), y)

指定 dim 后的返回值

# PyTorch 写法
result = torch.max(a, dim=1)

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

torch.max

torch.max(input,
          other,
          *,
          out=None)

paddle.maximum

paddle.maximum(x,
               y,
               name=None)

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

参数映射

PyTorch PaddlePaddle 备注
input x 输入的 Tensor ,仅参数名不一致。
other y 输入的 Tensor , 仅参数名不一致。
out - 表示输出的 Tensor , Paddle 无此参数,需要转写。

转写示例

out:指定输出

# 逐元素对比输入的两个 Tensor

# PyTorch 写法
torch.max(a, b, out=y)
# 在输入 other 时,比较 input 和 other 返回较大值

# Paddle 写法
paddle.assign(paddle.maximum(a, b), y)