異なるnumblockを持つ2つの配列の操作

Question

dask配列にtake_along_axisを実装しようとしています。

daskアレイAのブロックを取り込む操作と、daskアレイBの対応するブロックをマップする標準的な方法は何ですか？

A.numblocks != B.numblocksを使用している場合、再チャンクを使用する必要がありますか？

Answer 1

再チャンクは問題ありません。 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の通常の使用では、明示的に再チャンクを考える必要はありません。