在附加维度上重复keras（tensorflow）模型

作者: 易米烊光
发布时间: 2024-01-24 05:48:42 (6月前)
转自：

2 条回复

0#
回复此人
نسر الصحراء | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你可以这样做： </p> <pre> <code> input_ = Input((k, model.input.shape[1])) input_as_list = Lambda(lambda x: tf.unstack(x, axis=1))(input_) model_outputs = [model(x) for x in input_as_list] model_outputs = [Lambda(lambda x: K.expand_dims(x, axis=1))(y) for y in model_outputs] concat_output = Concatenate(axis=1)(model_outputs) new_model = Model(input_, concat_output) </code> </pre> </DIV>

编辑

登录后才能参与评论