jax.lax.max

jax.lax.max(x, y)

逐元素最大值: \(\mathrm{max}(x, y)\).

对于非复数输入，此函数直接降级为 stablehlo.maximum 操作。对于复数，这使用 (实数，虚数) 对上的字典序比较。

参数:

• x (ArrayLike) – 输入数组。必须具有匹配的 dtypes。如果两者都不是标量，则 x 和 y 必须具有相同的秩并且广播兼容。

• y (ArrayLike) – 输入数组。必须具有匹配的 dtypes。如果两者都不是标量，则 x 和 y 必须具有相同的秩并且广播兼容。

返回:

与 x 和 y 具有相同 dtype 的数组，包含逐元素最大值。

返回类型:

数组

另请参阅

• jax.numpy.maximum(): 更灵活的 NumPy 风格的最大值。

• jax.lax.reduce_max(): 沿数组轴的最大值。

• jax.lax.min(): 逐元素最小值。