我想计算张量每一行的加权和,权重是元素,另一个乘数是列号。示例失败:
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]
我该如何处理它?</p>
最佳答案
这就是内积的定义。与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])
关于python - 如何在 TensorFlow 中计算张量的加权和?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/48659449/