jax.numpy.amax

jax.numpy.amax(a, axis=None, out=None, keepdims=False, initial=None, where=None)

jax.numpy.max() 的別名。

參數：

• a (ArrayLike)

• axis (Axis | None)

• out (None | None)

• keepdims (bool)

• initial (ArrayLike | None | None)

• where (ArrayLike | None | None)

返回類型：

Array