Keras outputのスライスを次のlayerに渡したい

Question

前提・実現したいこと

あるlayerの出力のスライスを次のレイヤーに渡したい。

該当のソースコード

例えば次のモデルを定義します。

inputs = Input(shape=(64,))
x = Dense(16, activation='relu')(inputs)
outputs = Dense(1, activation='relu')(x)
model = Model(inputs, outputs)

上の例で、入力のshapeは次のように遷移します。
(None, 64)→(None, 16)→(None, 1)

ここで2回目のDense、すなわち(None, 16)→(None, 1)の部分では、16個のunitから1つのunitにfull connectしていますが、ここで16のユニットのうち前半4個だけを使いたいです。
すなわち入力のshapeを次のように遷移させたいです。
(None, 64)→(None, 16)→1つ前のlayerの前半(None, 4)→(None, 1)

これは質問用に簡略化したモデルなので、実際のモデルはもっと複雑なのですが、後半の(None, 8)もあとで使いたいと考えています。

よろしくお願いいたします。

Answer 1

tf.keras.layers.Lambda で解決

x = tf.keras.layers.Lambda(lambda t: t[:4])(x)