评价此页

torch.Tensor.unique#

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

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

请参见 torch.unique()