Tensorflow 从输入层屏蔽池层？
我需要在一个有遮罩的层中计算GlobalAveragePoolg1d。此遮罩作为一个层从模型输入。所以，我想知道这是否正确，它应该做什么Tensorflow 从输入层屏蔽池层？,tensorflow,keras,Tensorflow,Keras,我需要在一个有遮罩的层中计算GlobalAveragePoolg1d。此遮罩作为一个层从模型输入。所以，我想知道这是否正确，它应该做什么 input_ids = tf.keras.layers.Input(shape=(MAX_SEQ_lEN,), name='input_ids', dtype='int32') input_masks = tf.keras.layers.Input(shape=(MAX_SEQ_lEN,), name='input_masks', dtype='int32')
input_ids = tf.keras.layers.Input(shape=(MAX_SEQ_lEN,), name='input_ids', dtype='int32')
input_masks = tf.keras.layers.Input(shape=(MAX_SEQ_lEN,), name='input_masks', dtype='int32')
last_state=tf.keras.layers.LSTM(128, return_sequences=True)(input_ids )
avg=tf.keras.layers.GlobalAveragePooling1D()(last_state, mask=input_masks)
X = tf.keras.layers.Dense(1 , activation='sigmoid')(avg)
model = tf.keras.Model(inputs=[input_ids, input_masks], outputs = X)