torch.Tensor.unique_consecutive

Tensor.unique_consecutive(return_inverse=False, return_counts=False, dim=None)

消除每个连续的等效元素组中除第一个元素之外的所有元素。

参见 torch.unique_consecutive()