Mojo function

transpose

transpose[dtype: DType, //](output: TileTensor[dtype, linear_idx_type=output.linear_idx_type, element_size=output.element_size], input: TileTensor[dtype, linear_idx_type=input.linear_idx_type, element_size=input.element_size], perms: UnsafePointer[Int], ctx: Optional[DeviceContext] = None)

Permute the axis of input based on perms, and place the result in output.

Example:

transpose(output, input, [2, 0, 1])
# guarantees output[x, y, z] = input[z, x, y]

Parameters:

dtype (DType): The dtype of buffer elements.

Args:

output (TileTensor[dtype, linear_idx_type=output.linear_idx_type, element_size=output.element_size]): The output buffer.
input (TileTensor[dtype, linear_idx_type=input.linear_idx_type, element_size=input.element_size]): The input buffer.
perms (UnsafePointer[Int]): Permutation of the input axes.
ctx (Optional[DeviceContext]): The context to execute the work on.