jax.lax.argmin

Contents

jax.lax.argmin#

jax.lax.argmin(operand, axis, index_dtype)[source]#

Computes the index of the minimum element along axis.

Parameters:

operand (ArrayLike)
axis (int)
index_dtype (DTypeLike)

Return type: