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 struct
Axis
struct Axis
Wraps a tensor axis index, optionally normalizing negative values against the tensor rank.
Used by gather and scatter kernels to carry a validated axis through the call stack in a type-safe way.
Fields
- axis (
Int):
Implemented traits
AnyType,
Copyable,
Deinitable,
ImplicitlyCopyable,
Indexer,
Intable,
Movable,
RegisterPassable,
TrivialRegisterPassable
Methods
__init__
def __init__(axis: Int) -> Self
def __init__(out self, axis: Int, rank: Int)