会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
来一点音乐
让学习成为习惯
博客园
首页
新随笔
联系
订阅
管理
2019年9月24日
tensorflow2.0 squeeze出错
摘要: 用tf.keras写了自定义层,但在调用自定义层的时候总是报错,找了好久才发现问题所在,所以记下此问题。 问题代码 其中inputs的第一维为None,这里的代码为自定义的前向传播。我是想将得到的输出张量维度为1的维度删掉,因此调用了tf.squeeze方法,这时虽然没有报错但出现了问题。我分别打印
阅读全文
posted @ 2019-09-24 23:35 来一点音乐
阅读(495)
评论(0)
推荐(0)
编辑
公告