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!
😔 What went wrong?