jax.lax.argmin

内容

jax.lax.argmin#

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

计算沿 axis 的最小元素的索引。

参数:

  • operand (ArrayLike)

  • axis (int)

  • index_dtype (DTypeLike)

返回类型:

数组

内容