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)

返回类型:

Array