Mojo function
transpose_3d_swap_outer
transpose_3d_swap_outer[rank: Int, output_shape: DimList[output_shape.values], input_shape: DimList[input_shape.values], dtype: DType](output: NDBuffer[dtype, output.origin, output_shape], input: NDBuffer[dtype, input.origin, input_shape], perms: UnsafePointer[Scalar[DType.int], perms.origin], simplified_input_shape: IndexList[rank], simplified_rank: Int)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!