如何撤消 torch.unique_consecutive？

Question

如何有效地恢复 torch.unique_consecutive？ IE：

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



y = torch.SOMETHING(output,counts) #y equals x

Answer 1

请使用 torch.repeat_interleave(**args) 来完成您的任务

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