Skip to main content

Mojo struct

StaticTensorSpecList

struct StaticTensorSpecList[dtype: DType, rank: Int, //, internals_list: Variadic[StaticTensorSpecInternal[dtype, rank]], shapes_list: Variadic[IndexList[rank]], strides_list: Variadic[IndexList[rank]]]

A statically indexable list of data that can be assembled into a StaticTensorSpecList on demand. This handles the complexities that arise with heterogenous specs.

Implemented traitsโ€‹

AnyType, ImplicitlyDestructible

Methodsโ€‹

__getitem_param__โ€‹

__getitem_param__[index: Int](self, out result: StaticTensorSpec[dtype, rank, DimList.from_index_list[shapes_list[index]](), DimList.from_index_list[strides_list[index]]()])

Returns:

StaticTensorSpec

Was this page helpful?