умножение градиента матрицы

Я пытаюсь использовать массивы dask для обработки изображений на очень больших наборах данных. Часть этого - получение градиента, обработка и умножение их друг на друга. Однако это приводит к ошибке.

Минимальный пример без этапа обработки:

import dask.array as da
data = da.random.random((100, 100), chunks=(10, 10))
grad0, grad1 = da.gradient(data, axis=(0, 1))
(grad0 * grad1).compute()

Дает ошибку: ValueError: operands could not be broadcast together with shapes (8,12) (10,10).

Должно ли это работать?

Я использую dask 0.18.0 (самая последняя версия на PyPI).