更新除第 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]

但它可能比您当前的方法慢。