jax.Array.argmin

Contents

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:

Array