torch.repeat_interleave — PyTorch 1.10.1 documentation
pytorch.org › torchtorch.repeat_interleave(repeats, *, output_size=None) → Tensor. If the repeats is tensor ( [n1, n2, n3, …]), then the output will be tensor ( [0, 0, …, 1, 1, …, 2, 2, …, …]) where 0 appears n1 times, 1 appears n2 times, 2 appears n3 times, etc.