使用tensordot获取意外形状

Question

当我使用下面的代码在这两个张量之间进行tf.tensorbot时，我尝试在两个形状张量(2000，1，64) (2000，30，64)之间进行点积运算

test = tf.tensordot(enc_op,tf.transpose(query_with_time_axis),axes=1)

我得到的输出形状是(2000，30，1,2000)，但我必须得到(2000，30，1)的形状。

我这样做是为了注意力机制。有没有人能帮我。

Answer 1

尝试使用，

a=tf.transpose(a,perm=[0,2,1])

然后,

test=tf.matmul(b,a)

其中a是第一个张量，b是第二个张量