jax.numpy.nanargmin

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

返回指定轴上的最小值索引，忽略

LAX 后端实现 numpy.nanargmin().

警告：jax.numpy.argmin 对全 NaN 切片返回 -1，并且不引发错误。

原始文档字符串如下。

NaN。对于全 NaN 切片，会引发 ValueError。警告：如果切片仅包含 NaN 和 Inf，则无法信任结果。

参数：

• a (array_like) – 输入数据。

• axis (int, 可选) – 操作的轴。默认情况下使用扁平化输入。

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

• out (None | None)

返回值：

index_array – 索引数组或单个索引值。

返回类型：

ndarray