目录
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写作元组,传入