单张量预期时的张量列表

时间:2017-02-28 02:56:07

标签: tensorflow convolution

我使用concat来获取张量作为CNN的输入。但是得到了错误:单个Tensor预期时的张量列表

image_raw = img.tobytes()
image = tf.decode_raw(image_raw, tf.uint8)
image = tf.reshape(image, [1, image_height, image_width, 3])

image_val = image
for i in range(batch_size-1):
    image_val = tf.concat(0,[image_val,image])
return image_val

我已经搜索了这些问题的答案,添加    返回前image_val = tf.stack([image_val],0),但仍然得到同样的错误,为什么?

 **build  environment:**
 TensorFlow version 0.12
 python 3.5

2 个答案:

答案 0 :(得分:0)

错误List of Tensors when single Tensor expected来自您撰写tf.concat(0,[image_val,image])而非tf.concat([image_val,image],0)的事实。

答案 1 :(得分:0)

或许再次检查image_height, image_width的类型,因为有时需要将它们转换为整数dtype,例如tf.cast(image_height, tf.int32)