scalar_sqrt

功能说明

标量取开方操作：点击放大。

函数原型

scalar_sqrt(dst, src)

参数说明

表1 参数说明
参数名称	输入/输出	含义
dst	输出	目的操作数，数据类型需与源操作数保持一致。 Atlas 200/300/500 推理产品,，支持的数据类型为：Scalar(int64)。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64, float32)。 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	输入	源操作数。 Atlas 200/300/500 推理产品,，支持的数据类型为：Scalar(int64)、立即数(int64)。 Atlas 训练系列产品,，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas推理系列产品（Ascend 310P处理器）AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas推理系列产品（Ascend 310P处理器）Vector Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas A2训练系列产品/Atlas 800I A2推理产品，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas 200/500 A2推理产品，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。

支持的型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

Atlas推理系列产品（Ascend 310P处理器）AI Core

Atlas推理系列产品（Ascend 310P处理器）Vector Core

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

Atlas 200/500 A2推理产品

注意事项

支持负值，会先取abs，再sqrt。

返回值

无

调用示例

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为3
tik_instance.scalar_sqrt(dst_scalar, src_scalar)
tik_instance.BuildCCE(kernel_name="run_scalar_sqrt", inputs=[], outputs=[])

父主题： 单操作数