Hello. How can I tile/repeat a tensor in such a way that each value is tiled/repeated a different number of times other than looping then concatenating?
For e.g., from torch.Tensor([0, 1])
, how can I tile the first value two times and the second three times to get torch.Tensor([0, 0, 1, 1, 1])
. Thanks in advance!
torch.repeat_interleave
should work:
x = torch.tensor([0, 1])
print(torch.repeat_interleave(x, torch.tensor([2, 3])))
> tensor([0, 0, 1, 1, 1])
1 Like
Awesome thanks. I thought torch.repeat_interleave
worked for only integers.
1 Like