API reference
python:aix:aix.tensor.tensor_type
function tensor_type
tensor_type(element_type: astx.DataType, shape: tuple[int, ...]) -> astx.TensorType
Source: packages/aix/src/aix/tensor.py:120
Build one static-shape tensor surface type.
Parameters
-
element_type — Type: astx.DataType
-
shape — Type: tuple[int, Ellipsis]
Returns
Type: astx.TensorType