下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

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。

  • True:开启可读方式记录常量值。
  • False:不开启可读方式记录常量值。

返回值说明

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

约束说明

调用示例

1
2
3
import torch_npu, torchair
from torchair.ge import DataType
Const(0, dtype=DataType.DT_INT64, node_name='Const0')
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词