Source code for lacuna.array_api.searching

from .. import _dispatch as _dp
from .. import _namespace as _ns



[docs]
def argmax(x, axis=None, keepdims=False):
    return getattr(_ns, "argmax")(x, axis=axis, keepdims=keepdims)




[docs]
def argmin(x, axis=None, keepdims=False):
    return getattr(_ns, "argmin")(x, axis=axis, keepdims=keepdims)




[docs]
def nonzero(x):
    return getattr(_ns, "nonzero")(x)




[docs]
def count_nonzero(x, axis=None, keepdims=False):
    return _dp.count_nonzero(x, axis=axis, keepdims=keepdims)




[docs]
def where(cond, x, y):
    return getattr(_ns, "where")(cond, x, y)




[docs]
def searchsorted(x, v, *, side="left"):
    return getattr(_ns, "searchsorted")(x, v, side=side)