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 trait

Transformable

TransformAccess refined with the callable transform method.

Implemented traits

AnyType, Tensor, TransformAccess

comptime members

dtype

comptime dtype

rank

comptime rank

ShapeType

comptime ShapeType

Required methods

transform

def transform[width: Int, element_alignment: Int = Int(1)](self, idx: Coord, value: SIMD[Self.dtype, width]) -> SIMD[Self.dtype, width]

Applies the fused transform to a computed value at idx.

Parameters:

• ​width (Int): Number of elements in the value.
• ​element_alignment (Int): Assumed alignment in elements.

Args:

• ​idx (Coord): The element coordinate being produced.
• ​value (SIMD[_Self.dtype, width]): The computed SIMD[Self.dtype, width] value to transform.

Returns:

SIMD[_Self.dtype, width]: The transformed value; the parent kernel performs the final store.

shape

def shape(self) -> Self.ShapeType

Returns the tensor's runtime shape.

Returns:

_Self.ShapeType: The shape as a Coord of the argument's ShapeType.