[torch 参数更多] torch.Tensor.unique¶

torch.Tensor.unique¶

torch.Tensor.unique(sorted=True, return_inverse=False, return_counts=False, dim=None)

paddle.Tensor.unique¶

paddle.Tensor.unique(return_index=False, return_inverse=False, return_counts=False, axis=None, dtype='int64', name=None)

两者功能一致，torch 参数更多，具体如下：

参数差异¶

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | sorted | - | 是否返回前进行排序，Paddle 无此参数，需要转写。 | | return_inverse| return_inverse | 是否返回输入 Tensor 的元素对应在独有元素中的索引。 | | return_counts | return_counts | 是否返回每个独有元素在输入 Tensor 中的个数。 | | dim | axis | 选取的轴，仅参数名不一致。 | | - | return_index| 是否返回独有元素在输入 Tensor 中的索引，Pytorch 无此参数， Paddle 保持默认即可。|

转写示例¶

# 当 sorted 为‘True’时，torch 写法
torch.Tensor.unique(sorted=True, return_inverse=False, return_counts=False, dim=1)

# paddle 写法
paddle.Tensor.unique(return_index=False, return_inverse=False, return_counts=False, axis=1)

# 当 sorted 为‘False’时，暂时无法转写