scalar_sqrt

功能说明

标量取开方操作：。

函数原型

scalar_sqrt(dst, src)

参数说明

表1 参数说明
参数名称	输入/输出	含义
dst	输出	目的操作数，数据类型需与源操作数保持一致。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64, float32)。 Atlas 推理系列产品 AI Core，支持的数据类型为：Scalar(int64/float32)。 Atlas 推理系列产品 Vector Core，支持的数据类型为：Scalar(int64/float32)。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品，支持的数据类型为：Scalar(int64/float32)。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品，支持的数据类型为：Scalar(int64/float32)。 Atlas 200I/500 A2 推理产品，支持的数据类型为：Scalar(int64/float32)。
src	输入	源操作数。 Atlas 训练系列产品 ,，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas 推理系列产品 AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas 推理系列产品 Vector Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品，支持的数据类型为：Scalar(int64/float32)。 Atlas 200I/500 A2 推理产品，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。

支持的型号

Atlas 训练系列产品

Atlas 推理系列产品 AI Core

Atlas 推理系列产品 Vector Core

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

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas 200I/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=[])

父主题： 单操作数