[ 返回参数类型不一致 ]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)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
other | y | 输⼊ Tensor ,仅参数名不一致。 |