Function _array_like_safe

dask/array/utils.py:442–467

(np_func, da_func, a, like, **kwargs)

def _array_like_safe(np_func, da_func, a, like, **kwargs):
    from dask.array import Array

    if like is a and hasattr(a, "__array_function__"):
        return a

    if isinstance(like, Array):
        return da_func(a, **kwargs)

    if isinstance(a, Array) and is_cupy_type(a._meta):
        a = a.compute(scheduler="sync")

    if hasattr(like, "__array_function__"):
        return np_func(a, like=like, **kwargs)

    if type(like).__module__.startswith("scipy.sparse"):
        # e.g. scipy.sparse.csr_matrix
        kwargs.pop("order", None)
        if np.isscalar(a):
            a = np.array([[a]])
        return type(like)(a, **kwargs)

    # Unknown namespace with no __array_function__ support.
    # Quietly disregard like= parameter.
    # FIXME is there a use case for this?
    return np_func(a, **kwargs)


def array_safe(a, like, **kwargs):