为以下循环编写矢量化代码

Question

这是我们必须向量化的循环：

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

向量的维数如下：

y - num_train,1
X - num_train,3073
dW - 10,3073

我使用了以下高级索引方法，该方法似乎可以在Python解释器上正常运行，并附带一些小示例：

dW[y,] -= X[range(num_train),]

但这会错误地评估循环。