torch.Tensor.unique

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

返回输入张量的唯一元素。

请参阅 torch.unique()