summaryrefslogtreecommitdiff
path: root/numpy/_array_api/_searching_functions.py
blob: c6035ca7765b93e38b62e90fe9cc79d0c2f0af02 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def argmax(x, /, *, axis=None, keepdims=False):
    from .. import argmax
    return argmax(x, axis=axis, keepdims=keepdims)

def argmin(x, /, *, axis=None, keepdims=False):
    from .. import argmin
    return argmin(x, axis=axis, keepdims=keepdims)

def nonzero(x, /):
    from .. import nonzero
    return nonzero(x)

def where(condition, x1, x2, /):
    from .. import where
    return where(condition, x1, x2)