jax.lax.argmin

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

計算沿著 axis 的最小元素的索引。

參數：

• operand (ArrayLike)

• axis (int)

• index_dtype (DTypeLike)

回傳型別：

Array