pypto_pro.language.max#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

  • Atlas A3 训练系列产品/Atlas A3 推理系列产品：不支持

  • Atlas A2 训练系列产品/Atlas A2 推理系列产品：不支持

功能说明#

取两个标量中的较大值。

函数原型#

pypto_pro.language.max(
    lhs: Scalar,
    rhs: Scalar,
) -> Scalar

参数说明#

参数

输入/输出

说明

lhs

输入

左操作数，为整型或浮点型常量，或运行时整型或浮点型标量表达式。

rhs

输入

右操作数，为整型或浮点型常量，或运行时整型或浮点型标量表达式。

约束说明#

  • lhs和rhs必须是数值标量。两个操作数同为整型或同为浮点型时，结果提升为位宽较大的数据类型；一个为整型、另一个为浮点型时，两个操作数均转换为FP32后再比较。

  • 接口只接受两个位置参数，不支持关键字参数。

  • 适用范围：本接口仅用于标量取较大值。Tile逐元素取最大值请使用pypto_pro.language.maximum。

返回值说明#

返回两个标量中的较大值，Scalar类型。返回值数据类型由参数的数据类型提升结果决定。

调用示例#

基本用法#

import pypto_pro.language as pl

# 确保最小值
bottom_k = pl.max(1, trunk_len - topk + 1)

# 循环边界
end = pl.max(start, min_end)

# 标量变量之间
c = pl.max(a, b)

# 浮点标量
scale = pl.max(pl.const(2.5, pl.DT_FP32), pl.const(-1.5, pl.DT_FP32))