Class Constant

python/tvm/relax/expr.py:725–758

Constant Tensor Parameters ---------- data: tvm.runtime.Tensor The data of the constant tensor. struct_info: Optional[StructInfo] The struct info of the constant tensor. If not specified, infer it from data. span: Optional[Span] Span that points to orig

@tvm_ffi.register_object("relax.expr.Constant")
class Constant(ExprWithOp):
    """Constant Tensor

    Parameters
    ----------
    data: tvm.runtime.Tensor
        The data of the constant tensor.

    struct_info: Optional[StructInfo]
        The struct info of the constant tensor. If not specified, infer it from data.

    span: Optional[Span]
        Span that points to original source code

    Note
    ----
    Scalar constants are represented by ndim-0 constant tensors.
    """

    data: tvm.runtime.Tensor
    span: Span | None

    def __init__(
        self,
        data: tvm.runtime.Tensor,
        struct_info: StructInfo | None = None,
        span: Span | None = None,
    ) -> None:
        self.__init_handle_by_constructor__(
            _ffi_api.Constant,
            data,
            struct_info,
            span,  # type: ignore
        )


@tvm_ffi.register_object("relax.expr.Var")