dask数组梯度乘法

时间:2018-06-18 15:52:12

标签: python dask

我正在尝试使用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上的最新版本)。

1 个答案:

答案 0 :(得分:0)

该错误是由于dask.array中的错误引起的,并且已在开发分支https://github.com/dask/dask/pull/3647中进行了修复