Relay 常量#
tvm.relay.Constant
节点表示常量张量值。常量被表示为 NDArray
,允许 Relay 使用 TVM 算子来计算常量。
此节点还可以表示标量常数,因为标量是形状为 ()
的张量。因此,在文本格式中,数值和布尔字面值是将张量类型编码为零阶形状的常量的语法糖。
from tvm import nd, relay
const_a = nd.array(7)
const_op = relay.Constant(const_a)
str(const_op), repr(const_op)
('7i64', 'Constant(7)')