torch.unique_consecutive的反义词是什么?

What is the opposite of torch.unique_consecutive?

如何有效地还原 torch.unique_consecutive?即:

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

请使用 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]