NumPy使用更新参数进行缩减

时间:2017-05-03 15:16:54

标签: python python-3.x numpy vectorization

Xtheta分别为(m,k)(k,1)ndarray。请考虑以下代码

res = np.zeros(theta.shape)
for x_i in X:
    res +=  x_i @ theta

for周期可以重写为res = (X @ theta).sum()。现在考虑这个版本

for x_i in X:
    theta += x_i @ theta

有没有办法在没有for周期的情况下重写它,因为theta参数在每次迭代时都会更新?也许使用numpy.ufunc.reduce

0 个答案:

没有答案