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