Mojo function

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.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit