jax.lax.argmin# jax.lax.argmin(operand, axis, index_dtype)[源代码][源代码]# 计算沿 axis 的最小元素的索引。 参数: operand (ArrayLike) axis (int) index_dtype (DTypeLike) 返回类型: Array
