Keras：沿第4轴在两个4D张量上的点积

Question

我是张量数学的新手，所以这可能是一个愚蠢的问题。

我有两个张量，它们的尺寸均为(1,16,16,256)。 我的目标是在最后一个维度上应用点积，以便获得具有维度(1,16,16,1)的张量。 keras / tensorflow / numpy中是否有办法实现这一目标？

我尝试过

t_out = K.batch_dot(t_a,t_b, axes=[3, 3])

从Keras.backend API中

产生了维度(1,16,16,16)的输出。 我可能对函数的功能有误解。