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