Mojo struct
RuntimeTensorSpec
@register_passable(trivial)
struct RuntimeTensorSpec[type: DType, rank: Int]
Fields
- shape (
IndexList[rank]):
Implemented traits
AnyType,
Copyable,
Movable,
UnknownDestructibility
Methods
__init__
__init__(out self, spec: TensorSpec)
Construct from TensorSpec.
Args:
- spec (
TensorSpec): TensorSpec of given rank and type.
__getitem__
__getitem__(self, idx: Int) -> Int
get_tensor_spec
get_tensor_spec(self) -> TensorSpec
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!