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

Mojo module

transpose

The module implements Transpose functions.

Functions

• ​transpose: Permute the axis of input based on perms, and place the result in output.
• ​transpose_2d: Transposes the inner two dimensions of a simplified rank-2 tensor.
• ​transpose_3d_swap_inner: Transposes the inner two axes of a batched rank-3 tensor.
• ​transpose_3d_swap_outer: Transposes the outer two axes of a rank-3 tensor.
• ​transpose_4d_swap_middle: Transposes the middle two axes of a rank-4 tensor.
• ​transpose_inplace: Transposes a square rows x cols tile in place.
• ​transpose_strided: Transposes a tensor with arbitrary strides via a generic strided copy.
• ​transpose_trivial_memcpy: Copies the input buffer to the output buffer as a trivial transpose.