从输入层屏蔽池化层？

Question

我需要在具有蒙版的层中计算GlobalAveragePooling1D。此蒙版从输入到模型作为一个图层。因此，我想知道这样做是否正确？

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)