torch.Tensor.unique

Tensor.unique(sorted=True, return_inverse=False, return_counts=False, dim=None)[源代码]

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

参见 torch.unique()