tensorflowAPI-tf.contrib.data.Dataset from_tensor_slices()

tf.contrib.data.Dataset

函数

  • from_tensor_slices

Creates a Dataset whose elements are slices of the given tensors
创建一个数据集,其元素是切片张量

例子

train_imgs = tf.constant(['train/img1.png', 'train/img2.png',
                          'train/img3.png', 'train/img4.png',
                          'train/img5.png', 'train/img6.png'])
train_labels = tf.constant([0, 0, 0, 1, 1, 1])

tr_data = Dataset.from_tensor_slices((train_imgs, train_labels))

输出:
(b'train/img1.png', 0)
(b'train/img2.png', 0)
(b'train/img3.png', 0)
(b'train/img4.png', 1)
(b'train/img5.png', 1)
(b'train/img6.png', 1)

你可能感兴趣的:(tensorflowAPI-tf.contrib.data.Dataset from_tensor_slices())