如何计算Keras中的矢量点积？

Question

我希望在Keras中计算出矢量明智的点积.详细地说,我的意思是如果我有两个张量,A并且B两个都有形状(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])

但这不是我想要的.

谢谢!

Answer 1

您可以尝试以下方式:

import keras.backend as K

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