Output tensors to a Model must be the output of a TensorFlow `Layer`等问题记录

Output tensors to a Model must be the output of a TensorFlow Layer等问题记录

错误:Output tensors to a Model must be the output of a TensorFlow Layer
原因:tf.keras网络构造中使用到了非keras的网络层
解决方法:使用tf.keras的网络层替代,或将该操作写成函数,然后使用tf.keras.layer.Lambda层构造

函数:compute_output_shape(self, input_shape):
作用:keras中不确定输出形状是用来计算输出形状的

技巧:
把tensorflow的Dimension变为int型的方法
Diem.values
tensor_shape[0].value

tensorflow.contrib.framework.sort 在tensorflow2里面为tensorflow.sort

发布了143 篇原创文章 · 获赞 345 · 访问量 47万+

猜你喜欢

转载自blog.csdn.net/qq_31456593/article/details/89715281