更新除第 i 个条目之外的 numpy 数组

Question

我正在尝试实现 SGD 算法，其中有一个更新公式 

这可以通过使用轻松完成

temp = beta_old[i]
beta = beta_old
beta[i] = temp

但我觉得这很难看，我想知道是否有更优雅的方法来做到这一点（也许通过使用一些索引技巧）。

Answer 1

您可能想要使用面具：mask = np.ones(size, dtype=np.bool) mask[i] = false然后稍后使用掩码：beta[mask] = beta_old[mask]但它可能比您当前的方法慢。