gpt4 book ai didi

python - tensorflow 中的成对求和

转载 作者:太空宇宙 更新时间:2023-11-03 15:54:55 25 4
gpt4 key购买 nike

我有两个张量 AB,形状都是 [10 5]。如何计算形状为 [10 5 5] 的张量 C 使得 C[x, i, j] = A[x, i] + B [x, j]?

编辑:这是外积的和类比,而不是外积本身。

最佳答案

比@user1735003 的回答更具可读性和简洁性:

A[:, :, None] + B[:, None, :]

(实际上,另一个答案交换了坐标轴)

关于python - tensorflow 中的成对求和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44006511/

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