ValueError: too many values to unpack (expected 2)(tensorFlow的Dataset)

最新推荐文章于 2023-03-21 11:36:06 发布

LuckyFucky

最新推荐文章于 2023-03-21 11:36:06 发布

阅读量726

点赞数 1

文章标签： python tensorflow 深度学习 dataset

本文链接：https://blog.csdn.net/qq_43521665/article/details/115107040

版权

这篇博客讨论了在使用TensorFlow数据集时遇到的`ValueError: too many values to unpack(expected 2)`错误。问题在于尝试用两个变量`aa, bb`接收`c.map()`操作返回的单个元素。解决方法是通过遍历数据集来分别获取每个元素，如示例代码所示。修改后的代码成功地将映射操作应用于数据集并逐个处理结果。

摘要由CSDN通过智能技术生成

这个错误是因为返回的变量个数少于承接变量的个数

ValueError: too many values to unpack (expected 2)

出错代码如下，在用aa,bb装载返回值时报错，因为其实c返回的仍是一个完整的MapDataset

a=list([1,2,3,4,6])
b=list([1,2,3,4,6])
c=tf.data.Dataset.from_tensor_slices((a,b))
print(c)
def jisuan(x,y):
	return 2*x,y*2
aa,bb=c.map(lambda x,y:jisuan(x,y))

如果想要达到a,b分离的效果，需要对Dataset进行遍历，可以将代码改成下面的代码：

import tensorflow as tf

a=list([1,2,3,4,6])
b=list([1,2,3,4,6])
c=tf.data.Dataset.from_tensor_slices((a,b))
print(c)
def jisuan(x,y):
	return 2*x,y*2
cc=c.map(lambda x,y:jisuan(x,y))
for element in cc:
  a,b=element
  print(a)
# aa=cc.as_numpy_iterator()
# bb,dd=aa.next()
# print(bb)

效果：

tf.Tensor(2, shape=(), dtype=int32)
tf.Tensor(4, shape=(), dtype=int32)
tf.Tensor(6, shape=(), dtype=int32)
tf.Tensor(8, shape=(), dtype=int32)
tf.Tensor(12, shape=(), dtype=int32)

Process finished with exit code 0