Numpy":"运营商广播问题

时间:2018-02-27 15:18:09

标签: python numpy numpy-broadcasting

在下面的代码中,我写了2个方法,理论上(在我看来)应该做同样的事情。不幸的是,他们没有,我无法找出他们为什么不按照numpy文档做同样的事情。

if ann[len(ann)-1][0] == len(str):
                    break;

4 个答案:

答案 0 :(得分:2)

如上所述,由于缓冲在NumPy中的工作原理,原则上您不能在同一个操作中多次操作。为此目的,有at函数,可用于任何标准NumPy函数(addsubtract等)。对于您的情况,您可以这样做:

import numpy as np

dW = np.zeros((20, 10))
y = [1 for _ in range(100)]
X =  np.ones((100, 20))
# at modifies in place dW, does not return a new array
np.subtract.at(dW, (slice(None), y), X.T)

答案 1 :(得分:2)

这是this问题的列式版本。

那里的答案可以按照以下方式逐栏进行调整:

方法1:np.<ufunc>.at

>>> np.subtract.at(dW, (slice(None), y), X.T)

方法2:np.bincount

>>> m, n = dW.shape
>>> dW -= np.bincount(np.add.outer(np.arange(m) * n, y).ravel(), (X.T).ravel(), dW.size).reshape(m, n)

请注意,基于bincount的解决方案 - 即使涉及更多步骤 - 速度提高了约6倍。

>>> from timeit import repeat
>>> kwds = dict(globals=globals(), number=5000)
>>>
>>> repeat('np.subtract.at(dW, (slice(None), y), X.T); np.add.at(dW, (slice(None), y), X.T)', **kwds)
[1.590626839082688, 1.5769231889862567, 1.5802007300080732]
>>> repeat('_= dW; _ -= np.bincount(np.add.outer(np.arange(m) * n, y).ravel(), (X.T).ravel(), dW.size).reshape(m, n); _ += np.bincount(np.add.outer(np.arange(m) * n, y).ravel(), (X.T).ravel(), dW.size).reshape(m, n)', **kwds)
[0.2582490430213511, 0.25572817400097847, 0.25478115503210574]

答案 2 :(得分:1)

选项1:

for i in range(len(y)):
  dW[:, y[i]] -=  X[i]

这是有效的,因为您正在循环并更新上次更新的值。

选项2:

dW[:, [1,1,1,1,....1,1,1]] -=  [[1,1,1,1...1],
                                [1,1,1,1...1],
                                .
                                .
                                [1,1,1,1...1]]

它不起作用,因为更新同时发生在第一个索引并行而非串行方式。最初都是0,所以在-1中减去结果。

答案 3 :(得分:1)

我找到了解决这个问题的第三种解决方案。正常矩阵乘法:

ind = np.zeros((X.shape[0],dW.shape[1]))
ind[range(X.shape[0]),y] = -1
dW = X.T.dot(ind)

我在一些神经网络数据上使用上面提出的方法进行了一些实验。在我的示例中X.shape = (500,3073)W.shape = (3073,10)ind.shape = (500,10)

减法版本大约需要0.2秒(最慢)。矩阵乘法方法0.01秒(最快)。正常循环0.015然后bincount方法0.04秒。请注意,问题y是一个向量。这不是我的情况。只有一个的情况可以通过一个简单的总和来解决。