python - 如何计算 TensorFlow 中张量的加权和？

Question

我想计算张量每一行的加权和，权重是元素，另一个乘数是列号。示例自爆:

input:
[[0.2 0.5 0.3],
 [0.4 0.1 0.5]]

output:
[0.2*0+0.5*1+0.3*2, 0.4*0+0.1*1+0.5*2] = [1.1, 1.1]

我该怎么办？

Answer 1

这就是内积的定义。用numpy.dot :

data=[[0.2, 0.5, 0.3], [0.4, 0.1, 0.5]]
np.dot(data,range(len(data)+1))
array([ 1.1,  1.1])