jax.Array.max

内容

jax.Array.max#

abstract Array.max(axis=None, out=None, keepdims=False, initial=None, where=None)[source]#

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

有关完整文档,请参阅 jax.numpy.max()

参数:
  • self (Array)

  • axis (reductions.Axis)

  • out (None)

  • keepdims (bool)

  • initial (ArrayLike | None)

  • where (ArrayLike | None)

返回类型:

Array