tf.contrib.data.Dataset
函數(shù)
-
from_tensor_slices
Creates a Dataset whose elements are slices of the given tensors
創(chuàng)建一個(gè)數(shù)據(jù)集忧吟，其元素是切片張量
例子
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)
