jax.Array.argmin#

abstract Array.argmin(axis=None, out=None, keepdims=None)[源代码]#

返回最小值对应的索引。

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

参数:

  • self (Array)

  • axis (int | None)

  • out (None)

  • keepdims (bool | None)

返回类型:

数组