scalar_max

功能说明

比较并选取两个源操作数的最大值：

函数原型

scalar_max(dst, src0, src1)

参数说明

表1 参数说明
参数名称	输入/输出	含义
dst	输出	目的操作数。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64)。 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)。
src0	输入	源操作数0。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64)、立即数(int64)。 Atlas 推理系列产品 AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 Atlas 推理系列产品 Vector Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品，支持的数据类型为：Scalar(int64/float32)、立即数((int64)。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品，支持的数据类型为：Scalar(int64/float32)。 Atlas 200I/500 A2 推理产品，支持的数据类型为：Scalar(int64/float32)、立即数((int64)。
src1	输入	源操作数1。 Atlas 训练系列产品，支持的数据类型为：Scalar(int64)、立即数(int64)。 Atlas 推理系列产品 AI Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 Atlas 推理系列产品 Vector Core，支持的数据类型为：Scalar(int64/float32)、立即数(int64)。 Atlas A2 训练系列产品 / Atlas A2 推理系列产品，支持的数据类型为：Scalar(int64/float32)、立即数((int64)。 Atlas A3 训练系列产品 / Atlas A3 推理系列产品，支持的数据类型为：Scalar(int64/float32)。 Atlas 200I/500 A2 推理产品，支持的数据类型为：Scalar(int64/float32)、立即数((int64)。

支持的型号

Atlas 训练系列产品

Atlas 推理系列产品 AI Core

Atlas 推理系列产品 Vector Core

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

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

Atlas 200I/500 A2 推理产品

注意事项

操作数类型应保持一致。

返回值

None

调用示例

from tbe 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')
# dst_scalar=3
tik_instance.scalar_max(dst_scalar, src0_scalar, src1_scalar)
tik_instance.BuildCCE(kernel_name="run_scalar_max", inputs=[], outputs=[])

父主题： 双操作数