我使用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
答案 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)