Function func

dask/array/tests/test_atop.py:626–627

(x)

    x = da.ones((2, 2), chunks=(1, 1))

    def func(x):
        return np.stack([x, -x], axis=-1)

    y = da.blockwise(
        func,