Mojo function
prefix_product
prefix_product[: ImmutableOrigin, //, t: IntTuple[$0]](tuple: RuntimeTuple[t, element_type=element_type]) -> RuntimeTuple[prefix_product[::Origin[::Bool(t)]
Computes the prefix products of elements in the RuntimeTuple.
This function calculates the running product of elements, where each output element is the product of all previous elements in the input. This is commonly used in tensor computations to calculate stride values.
Parameters:
- t (
IntTuple[$0]): The IntTuple type parameter of the input RuntimeTuple.
Args:
- tuple (
RuntimeTuple[t, element_type=element_type]): The inputRuntimeTuple.
Returns:
A new RuntimeTuple containing the prefix products of the input elements.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!