Выровнять вместе с осью партии в TensorFlow / Keras

Question

В последовательной модели я пытаюсь перейти от формы вывода слоя (None, 300) к чему-то вроде (1,1,None*300), чтобы применить слой AveragePooling. На самом деле я хотел бы выровнять все (даже ось пакета), в то время как слои Flatten и Reshape всегда пропускают ось пакета. Есть идеи?

Answer 1

Вы можете использовать слой Lambda и K.reshape из бэкэнда следующим образом:

from keras import backend as K

out = Lambda(lambda x: K.reshape(x, (1, 1, -1)))(inp)