tf.data.Dataset.from_tensor_slices详解

本文详细介绍了tf.data.Dataset.from_tensor_slices的功能和用法。该方法用于从张量切片创建Dataset,沿第一个维度切片,可用于构建数据流水线。示例中展示了如何结合MNIST数据集使用此方法创建Dataset实例。

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

目录

tf.data.Dataset.from_tensor_slices的作用

tf.data.Dataset.from_tensor_slices的示例


tf.data.Dataset.from_tensor_slices的作用

tf.data.Dataset.from_tensor_slices方法可以创建一个由输入tensor的切片构成的Dataset实例。这个方法会沿着输入tensor的第1个维度进行切片。这个方法可以保留输入tensors的结构,去除每个tensor的第1个维度,并将其用作Dataset的维度。所有输入的tensor的第1个维度必须要一样

aDataset = tf.data.Dataset.from_tensor_slices(input_tensors, name='mydataset')

name参数是可选的。input_tensors可以是python列表,元组,字典,也可以是numpy array, 或者tensor。

tf.data.Dataset.from_tensor_slices的示例

如果已有数据的特征features和标签labels, 可以通过tf.data.Dataset.from_tensor_slices创建一个Dataset实例。以mnist数据集为例,展示2种创建方法。

方法1:直接将features和labels写作元组,传入

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

️Carrie爱学习

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值