jax.Array.argmax

abstract Array.argmax(axis=None, out=None, keepdims=None)

返回最大值的索引。

有关完整文档，请参阅 jax.numpy.argmax()。

参数：

• self (Array)

• axis (int | None)

• out (None)

• keepdims (bool | None)

返回类型：

Array