功能说明
对标量取绝对值操作：
。
函数原型
scalar_abs(dst, src)
参数说明
参数名称 |
输入/输出 |
含义 |
|---|---|---|
dst |
输出 |
目的操作数。 昇腾310 AI处理器，支持的数据类型为：Scalar(int64)。 昇腾910 AI处理器，支持的数据类型为：Scalar(int64)。 昇腾310P AI处理器AI Core，支持的数据类型为：Scalar(int64/float32)。 昇腾310P AI处理器Vector Core，支持的数据类型为：Scalar(int64/float32)。 |
src |
输入 |
源操作数，当src为Scalar时，要求src与dst数据类型保持一致。 昇腾310 AI处理器，支持的数据类型为：Scalar(int64)、立即数(int64)。 昇腾910 AI处理器，支持的数据类型为：Scalar(int64)、立即数(int64)。 昇腾310P AI处理器AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 昇腾310P AI处理器Vector Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 |
支持的芯片型号
昇腾310 AI处理器
昇腾910 AI处理器
昇腾310P AI处理器AI Core
昇腾310P AI处理器Vector Core
注意事项
无
返回值
无
调用示例
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=[])