Keras点产品层由最后一轴

Question

我有Siamese网络为图像制作特征图，我如何获得这些特征图与Keras的点积？

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

E.g。如果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)

这对你有用吗？