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