标量取开方操作:。
scalar_sqrt(dst, src)
参数名称 |
输入/输出 |
含义 |
---|---|---|
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)