Skip to main content
Log in

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]()]]