torch.unique_consecutive

本文主要是介绍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])

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])

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

这篇关于torch.unique_consecutive的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---