Const
函数功能
算子converter中的构图元素,表示一个Const节点,即图中的常量值。
该常量值在构建计算图时定义,且在整个图的执行过程中不会改变。
函数原型
def 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。
|
否 |
返回值说明
正常情况下,返回常量Tensor,否则失败报错。
约束说明
无
调用示例
1 2 3 |
import torch_npu, torchair from torchair.ge import DataType Const(0, dtype=DataType.DT_INT64, node_name='Const0') |
父主题: GE相关类和接口