最後の軸でケラスドットプロダクトレイヤー

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) 

など。 fe_net_aとfe_net_bの出力テンソルが形状（1,17,17,1024）の場合、テンソル形状（1,17,17）を生成する層を最終軸の内積をとって作成する方法は？

Answer 1

ケラスのドキュメントthe merge.Dot() functionから、これはあなたが探しているものですか？

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) 

これは機能しますか？