Warning

This page was created from a pull request.

jax.lax.argmin

jax.lax.argmin(operand, axis, index_dtype)

Computes the index of the minimum element along axis.

Parameters

• operand (Any) –

• axis (int) –

• index_dtype (Any) –

Return type

Tuple[Any, Any]