class tf.data.Dataset
- __init__
__init__()
- from_tensor_slices
@staticmethod
from_tensor_slices(tensors)
- 示例
input_data = [1, 2, 3, 4, 5]
dataset = tf.data.Dataset.from_tensor_slices(input_data)
- TextLineDataset
- TFRecordDataset
- make_one_shot_iterator
- map
- get_next
def parser(record):
features = ...
return features['feat1'], features['feat2']
dataset = tf.data.TFRecordDataset(input_files)
dataset = dataset.map(parser)
iterator = dataset.make_one_shot_iterator()
feat1, feat2 = iterator.get_next()
网友评论