python - 带有衰减的 Numpy 均匀分布

Question

我正在尝试构建一个均匀分布矩阵，每一行都以相同的速率衰减到 0。分布应该在 -1 和 1 之间。我正在看的是构建类似于以下内容的东西:

[[0.454/exp(0) -0.032/exp(1) 0.641/exp(2)...]
 [-0.234/exp(0) 0.921/exp(1) 0.049/exp(2)...]
 ...
 [0.910/exp(0) 0.003/exp(1) -0.908/exp(2)...]]

我可以使用以下方法构建均匀分布矩阵:

w = np.array([np.random.uniform(-1, 1, 10) for i in range(10)])

并且可以使用 for 循环实现预期的结果:

for k in range(len(w)):
    for l in range(len(w[0])):
        w[k][l] = w[k][l]/np.exp(l)

但想知道是否有更好的方法来完成此任务。

Answer 1

您可以使用 numpy 的广播功能来执行此操作:

w = np.random.uniform(-1, 1, size=(10, 10))
weights = np.exp(np.arange(10))
w /= weights