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

expand_strides

def 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.