mojo index详解及说明

index

Implements StaticIntTuple which is commonly used to represent N-D indices.

You can import these APIs from the utils package. For example:

from utils.index import StaticIntTuple

Aliases:

• ​mlir_bool = scalar<bool>

StaticIntTuple

A base struct that implements size agnostic index functions.

Parameters:

• ​size (Int): The size of the tuple.

Fields:

• ​data (StaticTuple[Int, size]): The underlying storage of the tuple value.

Implemented traits:

AnyType, Copyable, EqualityComparable, Movable, Sized, Stringable

Methods:

__init__

__init__() -> Self

Constructs a static int tuple of the given size.

Returns:

The constructed tuple.

__init__(value: index) -> Self

Constructs a sized 1 static int tuple of given the element value.

Args:

• ​value (index): The initial value.

Returns:

The constructed tuple.

__init__(elems: Tuple[Int, Int]) -> Self

Constructs a static int tuple given a tuple of integers.

Args:

• ​elems (Tuple[Int, Int]): The tuple to copy from.

Returns:

The constructed tuple.

__init__(elems: Tuple[Int, Int, Int]) -> Self

Constructs a static int tuple given a tuple of integers.

Args:

• ​elems (Tuple[Int, Int, Int]): The tuple to copy from.

Returns:

The constructed tuple.

__init__(elems: Tuple[Int, Int, Int, Int]) -> Self

Constructs a static int tuple given a tuple of integers.

Args:

• ​elems (Tuple[Int, Int, Int, Int]): The tuple to copy from.

Returns:

The constructed tuple.

__init__(*elems: Int) -> Self

Constructs a static int tuple given a set of arguments.

Args:

• ​elems (*Int): The elements to construct the tuple.

Returns:

The constructed tuple.

__init__(elem: Int) -> Self

Constructs a static int tuple given a set of arguments.

Args:

• ​elem (Int): The elem to splat into the tuple.

Returns:

The constructed tuple.

__init