【问题】KERAS中switch方法报错解决

最新推荐文章于 2020-06-30 21:19:11 发布

Laox1ao

最新推荐文章于 2020-06-30 21:19:11 发布

阅读量3.1k

点赞数 1

分类专栏：神经网络学习记录文章标签： python keras keras.backend.switch

本文链接：https://blog.csdn.net/Laox1ao/article/details/79007766

版权

神经网络学习记录专栏收录该内容

7 篇文章 0 订阅

订阅专栏

错误

ValueError: Shape must be rank 0 but is rank 3 for 'lambda_9/cond/Switch' (op: 'Switch') with input shapes: [?,20,30], [?,20,30].

错误原因

keras的backend（tensorflow）的switch方法会根据输入的维度是否是多维来选择调用tf.cond()或者tf.where()，但是在这个错误的log中发现，输入多维也在调用tf.cond()，原因是KERAS版本原因，错误出现在KERAS 2.0.5，升级到KERAS 2.1.2后解决。