API reference
python:astx:astx.tensor.TensorLiteral
class TensorLiteral
TensorLiteral(values: Sequence[astx.AST], *, element_type: astx.DataType, shape: Sequence[int], strides: Sequence[int] | None = None, offset_bytes: int = 0)
Source: packages/astx/src/astx/tensor.py:92
Internal Arrow C++ backed Tensor literal node.
Build one Arrow C++ tensor value from scalar values plus shape and stride metadata.
Attributes
-
values — Type: list[astx.AST]
-
element_type — Type: astx.DataType
-
shape — Type: tuple[int, Ellipsis]
-
strides — Type: tuple[int, Ellipsis] | None
-
offset_bytes — Type: int
-
type_ — Type: TensorType
Methods
-
__init__ — Initialize one Tensor literal.
-
get_struct — Return the structured representation of the Tensor literal.