ValueError: Shape must be rank 0 but is rank 1 for ‘Adam/update_weight/ApplyAdam‘ (op: ‘ApplyAdam‘)

最新推荐文章于 2020-09-04 18:33:22 发布

gyl2016

最新推荐文章于 2020-09-04 18:33:22 发布

阅读量1.3k

点赞数

分类专栏： tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/not_guy/article/details/107106959

版权

在使用TensorFlow时遇到ValueError，指出'Adam/update_weight/ApplyAdam'操作需要rank 0张量，但提供了rank 1的self.lr。原因是自定义的lr变量是形状为[1]的向量，而AdamOptimizer期望的是标量。解决方案是将lr变量改为标量。

摘要由CSDN通过智能技术生成

原代码：

self.lr = tf.placeholder(shape=[1],dtype=tf.float32,name="learning_rate")
。。。
。。。
        
optimizer = tf.train.AdamOptimizer(learning_rate=self.lr)
self.trainops = optimizer.minimize(self.cost)

报错信息：

InvalidArgumentError: Shape must be rank 0 but is rank 1 for 'Adam/update_weight/ApplyAdam' (op: 'ApplyAdam') with input shapes: [39,2], [39,2], [39,2], [], [], [1], [], [], [], [39,2].

During handling of the above exception, another exception occurred:
ValueError                                Traceback (most recent call last)
<ipython-input-149-f2a8d1d94ea1> in __init__(self)
     29 
     30         optimizer = tf.train.AdamOptimizer(learning_rate=self.lr)
---> 31         self.trainops = optimizer.minimize(self.cost)

出错原因：自