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 和 Infs，结果将不可信。

参数:

• a (array_like) – 输入数据。

• axis (int, optional) – 操作所沿的轴。默认使用展平的输入。

• keepdims (bool, optional) – 如果设置为 True，则被缩减的轴将作为大小为 1 的维度保留在结果中。通过此选项，结果将正确地与数组进行广播。

• out (None)

返回:

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

返回类型:

ndarray