Numpy渐变的大型3D阵列消耗所有内存

Question

我使用numpy来计算可能具有任意尺寸的大型3D数组的渐变

import numpy as np
N = 512
Nz = 512
result = np.zeros((N,N,Nz)).astype(np.float32)
gx, gy, gz = np.gradient(result)

问题是，由于ubuntu将大部分4GB RAM用于此算法，因此PC结果在几分钟内无法使用。

我尝试过cython但是得到了相同的结果。有什么比使用numpy渐变的蛮力方法更好吗？

提前致谢。