eager模式造成的bug

在 tensorflow1.x 版本中,可以通过命令 tf.enable_eager_execution() 开启eager模式,但是在该模式下,使用 tf.keras.layers.Input 会遇到问题

import tensorflow as tf
tf.enable_eager_execution()

x = tf.keras.layers.Input(shape=[2,10])
x_cut = x[:,0,:]
print(x_cut.shape)

得到的结果是

<unknown> 

由于程序不知道 x_cut 的维度,因此在调用 x_cut 的时候会出现报错。

而在非eager模式下可以得到正确结果

(?,10)

所以说,在 tensorflow1.x 中不要随便开启 eager 模式

posted @   Bill_H  阅读(171)  评论(1编辑  收藏  举报
编辑推荐:
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
阅读排行:
· 10年+ .NET Coder 心语 ── 封装的思维:从隐藏、稳定开始理解其本质意义
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
点击右上角即可分享
微信分享提示