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