Mojo function
vectorize
vectorize[*sizes: Int, *, dtype: DType, rank: Int, shape: DimList, _res_shape: DimList = __vectorize_shape[*::Int]($3)](buff: NDBuffer[dtype, rank, shape, strides]) -> Tuple[NDBuffer[dtype, rank, _res_shape], ElementLayout[rank, __to_static_tuple[*::Int,::Int]()]]
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!