尝试在2个数组之间进行点积时出现python内存错误

Question

我正在尝试使用softmax的导数https://medium.com/@aerinykim/how-to-implement-the-softmax-derivative-independently-from-any-loss-function-ae6d44363a9d

这是我使用的功能：

def softmax_grad(softmax):
    s = softmax.reshape(-1,1)
    return np.diagflat(s) - np.dot(s, s.T)

当我调用它时，我得到了这个错误

<ipython-input-72-4aed1c61658a> in softmax_prime(x)
      6 
      7     s = x.reshape(-1,1)
----> 8     return np.diagflat(s) - np.dot(s, s.T)

D:\anaconda\lib\site-packages\numpy\lib\twodim_base.py in diagflat(v, k)
    306     s = len(v)
    307     n = s + abs(k)
--> 308     res = zeros((n, n), v.dtype)
    309     if (k >= 0):
    310         i = arange(0, n-k)

MemoryError: 

是因为尺寸吗？我作为参数传递的数组大小为 (100, 60000)，点积在形状 (600000, 1) 之间 (1, 600000)

Answer 1

这将产生一个大小为 (600000, 600000) 的数组，我假设它不适合您的工作内存。