Kerasからバックエンドを使ってテンソルを集めるには？

Question

2次元numpy配列の入力を使ってKerasでモデルをコンパイルしようとしています。 私が必要とするのは、この2D配列のn番目の場所でベクトルを取得し、それをテンソル1Dテンソルとしてレイヤの1つに使用することです。 どうすればよいですか？ラムダ層はそれを行う必要があります使用

Answer 1

：

extracted_tensor = Lambda(lambda x: x[:,nth_index,:], output_shape=(1,dim_vector))(input) 
extracted_tensor = Flatten()(extracted_tensor) 

ノートxテンソル（ラムダ関数）で、あなたのアカウントにバッチ寸法を取るが、あなたはoutput_shapeパラメータでないこと。

私はこれが役に立ちそうです。