keras许多简单操作,都需要新建一个层,使用Lambda可以很好完成需求。
# 额外参数
def normal_reshape(x, shape):
return K.reshape(x,shape)
output = Lambda(normal_reshape, arguments={'shape':(-1, image_seq, 1000)})(output)
output = Lambda(lambda inp: K.mean(inp, axis=1), output_shape=(1000,))(output)