scalar_sqrt
功能说明
标量取开方操作：
。
函数原型
scalar_sqrt(dst, src)
参数说明
参数名称 |
输入/输出 |
含义 |
|---|---|---|
dst |
输出 |
目的操作数，数据类型需与源操作数保持一致。 Atlas 200/300/500 推理产品,，支持的数据类型为：Scalar(int64)。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64, float32)。 Atlas推理系列产品AI Core，支持的数据类型为：Scalar(int64/float32)。 Atlas推理系列产品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推理系列产品AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64/float32)。 Atlas推理系列产品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推理系列产品AI Core
Atlas推理系列产品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=[])