使用tf.convert_to_tensor 时遇到value error 怎么解决

在使用TensorFlow时,遇到在将包含PIL图片转为numpy数组的list转换为tensor时出现ValueError的问题。通过研究官方文档和试验,发现可能是list与array混用导致的维度不匹配。解决方案是将list+array转换为纯list或纯array,作者选择使用后者,通过强制转换成功解决问题。
摘要由CSDN通过智能技术生成

问题描述

最近写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.

探究

这么简单的函数怎么会出问题呢?

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值