更新时间:2021-03-18 GMT+08:00
分享

scalar_min

功能说明

比较并选取两个源操作数的最小值:

函数原型

scalar_min(dst, src0, src1)

参数说明

表1 参数说明

参数名称

输入/输出

含义

dst

输出

目的操作数,支持数据类型:Scalar(int64)。

src0

输入

源操作数0,支持数据类型:Scalar(int64)、立即数(int64)。

src1

输入

源操作数1,支持数据类型:Scalar(int64)、立即数(int64)。

注意事项

操作数类型应保持一致。

返回值

None

调用示例

from te import tik
tik_instance = tik.Tik()
src0_scalar = tik_instance.Scalar(dtype = "int64", name='src0_scalar', init_value=3)
src1_scalar = tik_instance.Scalar(dtype = "int64", name='src1_scalar', init_value=2)
dst_scalar = tik_instance.Scalar(dtype = "int64", name='dst_scalar')
tik_instance.scalar_min(dst_scalar, src0_scalar, src1_scalar)
分享:

    相关文档

    相关产品

close