Mojo function
transpose_4d_swap_middle
transpose_4d_swap_middle[rank: Int, type: DType, //](output: NDBuffer[type, rank, origin, shape], input: NDBuffer[type, rank, origin, shape, strides], perms: UnsafePointer[SIMD[index, 1]], simplified_input_shape: Index[rank], simplified_rank: Int)
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!