torch.unique_consecutive

就是将原始的torch去除重复的部分

原理

输入参数主要三个
x（必选）
return_inverse 【bool形】（是否返回输出对应数字在原torch中的下标）{可选}
return_counts【bool形】（是否返回输出对应数字的重复次数）{可选}
输出：
主要有x（返回去除重复元素的torch）
inverse
counts

案例

x = torch.tensor([1, 1, 2, 2, 2, 4, 3, 3,4, 1, 1, 2])
output = torch.unique_consecutive(x)
output
## tensor([1, 2, 4, 3, 4, 1, 2])
1
2
3
4

output, inverse_indices = torch.unique_consecutive(x, return_inverse=True)
inverse_indices
## tensor([0, 0, 1, 1, 1, 2, 3, 3, 4, 5, 5, 6])
1
2
3

output, counts = torch.unique_consecutive(x, return_counts=True)
counts
## tensor([2, 3, 1, 2, 1, 2, 1])
1
2
3