IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /docs/manual/basics.md). For the complete Mojo documentation index, see llms.txt.
Skip to main content
Version: 1.0.0b1
For the complete Mojo documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /docs/manual/basics.md).

expand_strides

expand_strides(shape: IntTuple, stride: Int) -> IntTuple

Expands a scalar stride into a stride tuple matching a shape tuple.

This function creates a stride tuple that matches the structure of a shape tuple, with each stride value calculated based on the cumulative product of shape dimensions.

Args:

  • ​shape (IntTuple): The shape tuple to match.
  • ​stride (Int): The base stride value to expand.

Returns:

IntTuple: A stride tuple matching the structure of the shape tuple.

View source