让X
和theta
分别为(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
?