jax.numpy.amax

内容

jax.numpy.amax#

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

jax.numpy.max() 的别名。

参数：:

  • a (ArrayLike)

  • axis (Axis | None)

  • out (None | None)

  • keepdims (bool)

  • initial (ArrayLike | None | None)

  • where (ArrayLike | None | None)

返回类型：:

数组

内容