[ 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 保持默认即可。 |
