Relay 常量

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)')