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?
Thank you! We'll create more content like this.
Thank you for helping us improve!