For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

repeat_interleave_kernel_shape

def repeat_interleave_kernel_shape(input: T, repeats: T, axis: Scalar) -> IndexList[T.rank]

Computes the output shape for the repeat_interleave graph op.

Args:

• ​input (T): Input tensor whose axis dimension is expanded by the repeat counts.
• ​repeats (T): One-dimensional integral tensor of per-element repeat counts, either size 1 or equal to input.dim(axis).
• ​axis (Scalar): Axis along which elements are repeated; negative values index from the last dimension.

Returns:

IndexList[T.rank]: The output shape with axis expanded by the repeat counts.