问题描述
最近写tensorflow 程序,使用PIL读取图片后转存为numpy 数组,然后创建batch数据时使用的是list,即,list元素是array,然而在将这个list转换为tensor时,出现了这样的问题:
....-got shape [64, 224, 224, 3], but wanted [64].
函数描述
convert_to_tensor(
value,
dtype=None,
name=None,
preferred_dtype=None
)
这个函数value接受的参数包括Tensor对象, numpy array, Python list, 和Python scalar.
探究
这么简单的函数怎么会出问题呢?