tf.data.Dataset.from_tensor_slices()

tf.data.Dataset.from_tensor_slices(tensors, name=None)

该函数的作用是接收tensor，对tensor的第一维度进行切分，并返回一个表示该tensor的切片数据集

# Slicing a 1D tensor produces scalar tensor elements.
import tensorflow as tf

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
print(dataset)
print(list(dataset.as_numpy_iterator()))

 

import tensorflow as tf

# Slicing a 2D tensor produces 1D tensor elements.
dataset = tf.data.Dat