pypto_pro.language.const#

产品支持情况#

  • Ascend 950PR/Ascend 950DT:支持

  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持

  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持

功能说明#

创建指定数据类型的编译期常量标量。用于需要显式指定类型的标量值场景,如与pypto_pro.language.DT_INT32类型的Tile做比较、初始化索引值等。

函数原型#

result = pypto_pro.language.const(value, dtype)

参数说明#

参数

输入/输出

说明

value

输入

数值常量,支持Python int或float常量值。

dtype

输入

目标数据类型,支持pypto_pro.language.DT_INT8、pypto_pro.language.DT_INT16、pypto_pro.language.DT_INT32、pypto_pro.language.DT_INT64、pypto_pro.language.DT_UINT8、pypto_pro.language.DT_UINT16、pypto_pro.language.DT_UINT32、pypto_pro.language.DT_UINT64、pypto_pro.language.DT_FP16、pypto_pro.language.DT_FP32、pypto_pro.language.DT_BF16、pypto_pro.language.DT_BOOL等。

约束说明#

无。

返回值说明#

标量值(Scalar),类型由dtype指定。

调用示例#

import pypto_pro.language as pl
# 创建 INT64 类型的零常量
zero_idx = pl.const(0, pl.DT_INT64)

# 创建 INT32 类型的常量
offset = pl.const(42, pl.DT_INT32)

# 创建 FP16 类型的常量
scale = pl.const(1.0, pl.DT_FP16)

# 创建负值常量
neg_offset = pl.const(-1, pl.DT_INT32)