jax.Array.argmin — JAX documentation (original) (raw)

jax.Array.argmin#

abstract Array.argmin(axis=None, out=None, keepdims=None)[source]#

Return the index of the minimum value.

Refer to jax.numpy.argmin() for the full documentation.

Parameters:

self (Array)
axis (int | None)
out (None)
keepdims (bool | None)

Return type: