Skip to main content
Log in

Mojo struct

StaticTensorSpec

Fields

  • shape (StaticIntTuple[rank]):

Implemented traits

AnyType, Copyable, Movable

Methods

__init__

__init__(inout self: Self, spec: TensorSpec)

Construct from TensorSpec.

Args:

  • spec (TensorSpec): TensorSpec of given rank and type.

__getitem__

__getitem__(self: Self, idx: Int) -> Int

get_tensor_spec

get_tensor_spec(self: Self) -> TensorSpec

Was this page helpful?