tensorflow2.0生成dataset的几点坑

最新推荐文章于 2024-10-21 13:59:27 发布

qq_903012463

最新推荐文章于 2024-10-21 13:59:27 发布

阅读量388

点赞数

分类专栏： tensorflow应用

本文链接：https://blog.youkuaiyun.com/qq_43309133/article/details/114682691

版权

tensorflow应用专栏收录该内容

37 篇文章

订阅专栏

本文介绍如何使用TensorFlow的tf.data.Dataset.from_tensor_slices方法构建训练数据集，并解释了常见错误及解决办法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

(x, y), (x_test, y_test) = keras.datasets.mnist.load_data()
db_train = tf.data.Dataset.from_tensor_slices((x, y)）

上面x是numpy.array数组。
tf.data.Dataset.from_tensor_slices还可以直接接收列表。
有一种情况会报错，ValueError: Failed to convert a NumPy array to a Tensor (Unsupported object type numpy.ndarray).
这是因为数组里面有维度没展开。比如x.shape ==> (64,) 然而每个元素又是(24,5)。这样就不行，[64,24,5]就可以。