jax.numpy.argmin

jax.numpy.argmin(a, axis=None, out=None, keepdims=None)

沿轴返回最小值索引。

numpy.argmin() 的 LAX 后端实现。

原始文档字符串如下。

参数：

• a (array_like) – 输入数组。

• axis (int, 可选) – 默认情况下，索引指向扁平化的数组，否则沿指定的轴。

• keepdims (bool, 可选) – 如果设置为 True，则减少的轴将作为大小为一的维度保留在结果中。使用此选项，结果将针对数组正确广播。

• out (None | None)

返回值：

index_array – 数组的索引数组。它具有与 a.shape 相同的形状，但去掉了沿 axis 的维度。如果 keepdims 设置为 True，则 axis 的大小将为 1，结果数组将与 a.shape 具有相同的形状。

返回类型：

整数类型的 ndarray