Const

功能说明

算子Converter中的构图元素,表示一个Const节点,即图中的常量值。该常量值在构建计算图时定义,且在整个图的执行过程中不会改变。

函数原型

Const(v: Any, dtype: int = None, node_name=None, readable=True) -> Tensor

参数说明

参数 输入/输出 说明
v 输入 设定的常量值,支持任意数据类型,例如1.0。
dtype 输入 常量值的数据类型,默认为整型,类型取值参见DataType类
node_name 输入 常量节点名,默认为None,例如'const_1',同一张图中节点名不允许重复。
readable 输入 是否在图上增加属性以可读的方式记录const值,默认为True。
True:开启可读方式记录常量值。
False:不开启可读方式记录常量值。

返回值说明

正常情况下,返回常量Tensor,否则失败报错。

约束说明

调用示例

import torch_npu, torchair
from torchair.ge import DataType, Const
Const(0, dtype=DataType.DT_INT64, node_name='Const0')