jax.numpy.nanargmax

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

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

LAX-backend 实现的 numpy.nanargmax()。

警告：jax.numpy.argmax 对所有 NaN 切片返回 -1，并且不会引发错误。

原始文档字符串如下。

NaN。对于全为NaN的切片，会引发 ValueError。警告：如果切片仅包含NaN和-Infs，则结果不可信。

参数:

• a (array_like) – 输入数据。

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

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

• out (None)

返回:

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

返回类型:

ndarray