Ну, вы можете нарезать их по своему усмотрению, например, как нарезать массив или список Python, и использовать K.concatenate
, все в слое Lambda
.Например:
from keras import backend as K
# ...
out = Lambda(lambda x: K.concatenate([x[0][:,:10],
x[1][:,:10],
x[2][:,:10]], axis=the_concat_axis))([conv1, conv2, pool])
Обратите внимание, что первая ось является осью пакета, поэтому вы можете сохранить ее все (т.е. используйте :
, как указано выше).