the given axis.
Return type:
      Tensor
Example:
  >>> x = torch.tensor([1, 2, 3])
  >>> x.repeat_interleave(2)
  tensor([1, 1, 2, 2, 3, 3])
  >>> y = torch.tensor([[1, 2], [3, 4]])
  >>> torch.repeat_interleave(y, 2)
  tensor([1, 1, 2, 2, 3, 3, 4, 4])
  >>> torch.repeat_interleave(y, 3, dim=1)
  tensor([[1, 1, 1, 2, 2, 2],
          [3, 3, 3, 4, 4, 4]])
  >>> torch.repeat_interleave(y, torch.tensor([1, 2]), dim=0)
  tensor([[1, 2],
          [3, 4],
          [3, 4]])
  >>> torch.repeat_interleave(y, torch.tensor([1, 2]), dim=0, output_size=3)
  tensor([[1, 2],
          [3, 4],
          [3, 4]])

torch.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.