按最后一个轴的 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

来自 the merge.Dot() function 的 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)

那对你有用吗 ？