Mojo function
expand_strides
expand_strides(shape: IntTuple[origin], 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[origin]): The shape tuple to match. - stride (
Int): The base stride value to expand.
Returns:
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!