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!
If you'd like to share more information, please report an issue on GitHub

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other