我想问一下是否有办法 reshape Fortran-contiguous (column-major) order中的dask
数组。因为尚不支持 np.reshape
函数的并行版本 (see here)。
最佳答案
Fortran 连续(列优先)顺序只是 C 连续(行优先)顺序的倒转。因此,对于 dask 数组不支持 order='F'
的事实,有一个简单的解决方法:
- 转置数组以反转其维度。
- 将其重新塑造成您想要的形状的反面。
- 将其转回。
在函数中:
def reshape_fortran(x, shape):
return x.T.reshape(shape[::-1]).T
用 NumPy/dask 转置基本上是免费的(它不复制任何数据),所以原则上这个操作也应该是相当高效的。
这是一个简单的测试来验证它是否正确:
In [48]: import numpy as np
In [49]: import dask.array as da
In [50]: x = np.arange(100).reshape(10, 10)
In [51]: y = da.from_array(x, chunks=5)
In [52]: shape = (2, 5, 10)
In [53]: np.array_equal(reshape_fortran(y, shape).compute(),
...: x.reshape(shape, order='F'))
...:
Out[53]: True
关于python - 按 Fortran 连续顺序 reshape dask.array,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45479325/