gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-11-28 21:38:06 25 4
gpt4 key购买 nike

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

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]

我该怎么办?

最佳答案

这就是内积的定义。用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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com