jax.lax.reduce_max#

jax.lax.reduce_max(operand, axes)[源代码]#

计算一个或多个数组轴上的元素的最大值。

参数:

  • operand (ArrayLike) – 用于计算最大值的数组。

  • axes (Sequence[int]) – 零个或多个唯一整数的序列，指定要减少的轴。 每个条目必须满足 0 <= axis < operand.ndim。

返回:

与 operand 具有相同 dtype 的数组，其形状对应于 operand.shape 的维度，并移除了 axes。

返回类型:

数组

另请参阅