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: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec], repeats: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=repeats.static_spec], axis: Scalar) -> IndexList[input.static_spec.rank]

Computes the output shape for the repeat_interleave graph op.

Args:

• ​input (ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec]): Input tensor whose axis dimension is expanded by the repeat counts.
• ​repeats (ManagedTensorSlice[IOSpec[_, _].Input, static_spec=repeats.static_spec]): 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[input.static_spec.rank]: The output shape with axis expanded by the repeat counts.