Keras点积层（按最后一个轴）

Question

我有暹罗网络，可以为图像生成特征图，如何使用Keras获得这些特征图的点积？

input_a = Input(input_size)
input_b = Input(input_size)
fe_net_a = model(input_a)
fe_net_b = model(input_b)

例如，如果输出张量为fe_net_a且fe_net_b具有形状（1、17、17、1024），则如何创建将产生形状为（1、17、17）的张量的层，并取最后一个轴的点积？

Answer 1

从该merge.Dot()函数的 keras 文档中，这是您要寻找的吗？

from keras.layers import Input, dot

input_a = Input(input_size)
input_b = Input(input_size)
fe_net_a = model(input_a)
fe_net_b = model(input_b)

fe_ab = dot([fe_net_a,fe_net_b], axes = -1)

那对你有用吗 ？