Argument must be a dense tensor: range(0, 3) - got shape [3], but wanted []

ygfrancois

于 2018-06-14 09:42:52 发布

阅读量3.9k

点赞数 2

文章标签： tensorflow python3

本文解决了一个在使用TensorFlow过程中遇到的ValueError问题，该问题是由于从Python2升级到Python3后，range()函数的行为变化导致的。具体表现为尝试将range(0,3)转换为张量时失败，因为期望的是空形状，而实际得到了一个范围对象。通过将range()替换为list(range())成功解决了这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tensorflow遇到问题：

ValueError: Tried to convert 't' to a tensor and failed. Error: Argument must be a dense tensor: range(0, 3) - got shape [3], but wanted [].

原因是python2转python3后，range（）返回的shape是range（0，3）而不再是list。

需要将models/research/object_detection/utils/learning_schedules.py里的：

  rate_index = tf.reduce_max(tf.where(tf.greater_equal(global_step, boundaries),
                                      range(num_boundaries),
                                      [0] * num_boundaries))

修改成：

  rate_index = tf.reduce_max(tf.where(tf.greater_equal(global_step, boundaries),
                                      list(range(num_boundaries)),
                                      [0] * num_boundaries)