[ 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 无此参数。可直接删除,PyTorch 目前忽略了此参数。
return_inverse return_inverse 是否返回输入 Tensor 的元素对应在独有元素中的索引。
return_counts return_counts 是否返回每个独有元素在输入 Tensor 中的个数。
dim axis 选取的轴,仅参数名不一致。
- return_index 是否返回独有元素在输入 Tensor 中的索引,PyTorch 无此参数, Paddle 保持默认即可。