ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`

最新推荐文章于 2022-03-16 16:48:35 发布

gyl2016

最新推荐文章于 2022-03-16 16:48:35 发布

阅读量3.3k

点赞数

分类专栏： tensorflow Bug 深度学习

本文链接：https://blog.csdn.net/not_guy/article/details/108603694

版权

在学习tensorflow.keras的多输入多输出时遇到错误：ValueError: Output tensors to a Model must be the output of a TensorFlow `Layer`。错误源于中间操作没有Keras层的历史信息。解决方案是使用tf.keras.layers.Lambda封装tf.keras.backend.concatenate函数。参考了相关博客文章进行问题解决。

摘要由CSDN通过智能技术生成

最近在看tensorflow.keras的多输入多输出时，碰到了这个问题，在这里记录一下。

首先给出报错代码：

# 两个输入
main_input = tf.keras.layers.Input(shape=(784,),dtype=tf.float32,name="main_input")
auxiliary_input = tf.keras.layers.Input(shape=784,),dtype=tf.float32,name="auxiliary_input")

## 网络1
x = tf.keras.layers.Dense(128,activation="relu")(main_input)
dense_out = tf.keras.layers.Dense(10,activation="softmax")(x)
output1 = tf.keras.layers.Dense(10,activation="softmax")(dense_out)    ## 输出1

## 网络2
x = tf.keras.backend.concatenate([dense_out, auxiliary_input])
x = tf.keras.layers.Dense(128,activation="relu")(x)
x = tf.keras.layers.Lambda(tf.keras.layers.Dropout(0.5))(x)
x = tf.keras.layers.Dense(128,activation="relu")(x)
main_output = tf.keras.layers.Dense(10,activation="softmax")(x)  # 输出2

model = tf.keras