function tensor_shape

tensor_shape(data_type: astx.DataType | None) -> tuple[int, ...] | None

Source: packages/aix/src/aix/tensor.py:106

Return the declared tensor shape when available.

Parameters

• data_type — Type: astx.DataType | None

Returns

Type: tuple[int, Ellipsis] | None