gpt4 book ai didi

tensorflow - 如何在 Keras 中计算向量点积?

转载 作者:行者123 更新时间:2023-12-02 07:33:43 24 4
gpt4 key购买 nike

我希望在 Keras 中计算向量明智的点积。详细来说,我的意思是,如果我有两个张量 AB,形状均为 (None, 30, 100),我想计算结果 C 的形状 (None, 30, 1) 满足

C[:,:,i] = dot(A[:,:,i], B[:,:,i]).

我想知道这是否可能,因为后端的 batch_dot() 函数只会返回形状 (None, 30, 30) 并且只有关系

C[:,i,j] = dot(A[:,:,i], B[:,:,j])

但这不是我想要的。

谢谢!

最佳答案

您可以尝试以下操作:

import keras.backend as K

C = K.sum(A * B,axis=-1,keepdims=True)

关于tensorflow - 如何在 Keras 中计算向量点积?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/43853741/

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