Function mydiff

dask/array/tests/test_cupy_gufunc.py:16–17

(x)

def test_apply_gufunc_axis():
    def mydiff(x):
        return np.diff(x)

    a = cupy.random.default_rng().standard_normal((3, 6, 4))
    da_ = da.from_array(a, chunks=2, asarray=False)