python - 在 Numpy 中编写从 i = 1 到 n, log(1 + exp(w_i)) 的和的有效方法是什么

Question

令w = (w_1, w_2, w_3, ...., w_n)为一个数组，n很大

不使用循环，我想定义函数

sum from i = 1 to i = n, log(1 + exp(w_i))

Numpy 中有处理这个问题的向量运算吗？我在想

np.dot(np.ones((n,)), np.log(1+np.exp(w))

但我不知道这是否有效。

Answer 1

您可以使用np.sum(...)对数组的所有元素求和。

虽然 np.log(1+np.exp(w)) 应该可以正常工作，但还有 np.log1p(...)在数字非常小的情况下，它可以更精确地计算一加值的 ln。

把它们放在一起:

result = np.sum(np.log1p(np.exp(w)))