For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

transpose_3d_swap_outer

def transpose_3d_swap_outer[rank: Int, dtype: DType, //](output: TileTensor[dtype, Storage=output.Storage, linear_idx_type=output.linear_idx_type], input: TileTensor[dtype, Storage=input.Storage, linear_idx_type=input.linear_idx_type], perms: Pointer[Int, _safe=False], simplified_input_shape: IndexList[rank], simplified_rank: Int)

Transposes the outer two axes of a rank-3 tensor.

Maps an MxNxK input to an NxMxK output by delegating to the rank-4 middle-swap helper with an implicit outer dimension of size 1.

Parameters:

rank (Int): Number of dimensions in the simplified_input_shape array (inferred).
dtype (DType): Element type of the input and output buffers (inferred).

Args:

output (TileTensor[dtype, Storage=output.Storage, linear_idx_type=output.linear_idx_type]): The output buffer with the outer two axes swapped.
input (TileTensor[dtype, Storage=input.Storage, linear_idx_type=input.linear_idx_type]): The input buffer.
perms (Pointer[Int, _safe=False]): Permutation of the input axes.
simplified_input_shape (IndexList[rank]): Shape of the tensor after simplification.
simplified_rank (Int): Effective rank after simplification.