scalar_sqrt

功能说明

标量取开方操作:

函数原型

scalar_sqrt(dst, src)

参数说明

表1 参数说明

参数名称

输入/输出

含义

dst

输出

目的操作数,数据类型需与源操作数保持一致。

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

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

src

输入

源操作数。

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

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

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

注意事项

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

返回值

调用示例

from tbe import tik
tik_instance = tik.Tik()
src_scalar = tik_instance.Scalar(dtype = "int64")
src_scalar.set_as(10)
dst_scalar = tik_instance.Scalar(dtype = "int64")
tik_instance.scalar_sqrt(dst_scalar, src_scalar)