repeat

dragon.vm.torch.repeat(
  input,
  sizes
)[source]

Repeat elements along the specified dimensions.

Parameters:
  • input (dragon.vm.torch.Tensor) – The input tensor.
  • sizes (Sequence[int]) – The number of times to repeat.
Returns:

dragon.vm.torch.Tensor – The output tensor.