tf.data.Dataset.from_tensor_slices详解

目录

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, 或者tens