torch.Tensor.unique

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

傳回輸入張量的唯一元素。

請參閱 torch.unique()