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  阅读(170)  评论(1编辑  收藏  举报
编辑推荐:
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
阅读排行:
· 新年开篇:在本地部署DeepSeek大模型实现联网增强的AI应用
· DeepSeek火爆全网,官网宕机?本地部署一个随便玩「LLM探索」
· Janus Pro:DeepSeek 开源革新,多模态 AI 的未来
· 互联网不景气了那就玩玩嵌入式吧,用纯.NET开发并制作一个智能桌面机器人(三):用.NET IoT库
· 上周热点回顾(1.20-1.26)
点击右上角即可分享
微信分享提示