文档
注册
评分
提单
论坛
小AI

scalar_abs

功能说明

对标量取绝对值操作:

函数原型

scalar_abs(dst, src)

参数说明

表1 参数说明

参数名称

输入/输出

含义

dst

输出

目的操作数。

Atlas 200/300/500 推理产品,支持的数据类型为:Scalar(int64)。

Atlas 训练系列产品,支持的数据类型为:Scalar(int64)。

Atlas推理系列产品(Ascend 310P处理器)AI Core,支持的数据类型为:Scalar(int64/float32)。

Atlas推理系列产品(Ascend 310P处理器)Vector Core,支持的数据类型为:Scalar(int64/float32)。

Atlas A2训练系列产品/Atlas 800I A2推理产品,支持的数据类型为:Scalar(int64/float32)。

Atlas 200/500 A2推理产品,支持的数据类型为:Scalar(int64/float32)。

src

输入

源操作数,当src为Scalar时,要求src与dst数据类型保持一致。

Atlas 200/300/500 推理产品,支持的数据类型为:Scalar(int64)、立即数(int64)。

Atlas 训练系列产品,支持的数据类型为:Scalar(int64)、立即数(int64)。

Atlas推理系列产品(Ascend 310P处理器)AI Core,支持的数据类型为:Scalar(int64/float32)、立即数(int64)。

Atlas推理系列产品(Ascend 310P处理器)Vector Core,支持的数据类型为:Scalar(int64/float32)、立即数(int64)。

Atlas A2训练系列产品/Atlas 800I A2推理产品,支持的数据类型为:Scalar(int64/float32)、立即数(int64)。

Atlas 200/500 A2推理产品,支持的数据类型为:Scalar(int64/float32)、立即数(int64)。

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品(Ascend 310P处理器)AI Core

Atlas推理系列产品(Ascend 310P处理器)Vector Core

Atlas A2训练系列产品/Atlas 800I A2推理产品

Atlas 200/500 A2推理产品

注意事项

返回值

调用示例

from tbe import tik
tik_instance = tik.Tik()
src_scalar = tik_instance.Scalar(dtype="int64")
dst_scalar = tik_instance.Scalar(dtype="int64")
# src_scalar=-10
src_scalar.set_as(-10)
# dst_scalar=10
tik_instance.scalar_abs(dst_scalar, src_scalar)
tik_instance.BuildCCE(kernel_name="run_scalar_abs", inputs=[], outputs=[])
搜索结果
找到“0”个结果

当前产品无相关内容

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