torch.Tensor.Tensor.repeat_interleave
Tensor.repeat_interleave(repeats: number): Tensor<DynamicShape, D, Dev>Tensor.repeat_interleave(repeats: Tensor, dim: number, output_size: number, options: RepeatInterleaveOptions): Promise<Tensor>Repeats elements of a tensor along a dimension.
Parameters
repeatsnumber- Number of repetitions for each element
Returns
Tensor<DynamicShape, D, Dev>– Tensor with repeated elements (sync if repeats is number, async if repeats is Tensor)