jax.Array.argmax#

抽象 Array.argmax(axis=None, out=None, keepdims=None)[source]#

返回最大值的索引。

参考 jax.numpy.argmax() 以获取完整文档。

参数:

  • self (Array)

  • axis (int | None)

  • out (None)

  • keepdims (bool | None)

返回类型:

Array