jax.Array.max

abstract Array.max(axis=None, out=None, keepdims=False, initial=None, where=None)

返回沿给定轴的数组元素的最大值。

请参考 jax.numpy.max() 以获取完整文档。

参数:

• self (Array)

• axis (reductions.Axis)

• out (None)

• keepdims (bool)

• initial (ArrayLike | None)

• where (ArrayLike | None)

返回类型:

Array