在具有不同numblock的两个阵列上操作

Question

我正在尝试为dask数组实现take_along_axis。

映射从dask数组A获取块的操作以及来自dask数组B的相应块的标准方法是什么？

A.numblocks != B.numblocks时我应该使用重新发送吗？

Answer 1

Rechunking很好。 Dask.array算法内部相当频繁地对齐块。如果它们适合您，您也可以考虑使用map_blocks或atop等功能。 Dask开发人员优化了这些。

• http://dask.pydata.org/en/latest/array-api.html#dask.array.core.atop
• http://dask.pydata.org/en/latest/array-api.html#dask.array.core.map_blocks

如果您构建自己的算法，这些只是重要的。正常使用dask.arrays并不需要明确考虑重新组合。