Module: tf.data

TensorFlow 2 version

tf.data.Dataset API for input pipelines.

See Importing Data for an overview.

Modules

experimental module: Experimental API for building input pipelines.

Classes

class Dataset: Represents a potentially large set of elements.

class DatasetSpec: Type specification for tf.data.Dataset.

class FixedLengthRecordDataset: A Dataset of fixed-length records from one or more binary files.

class Iterator: Represents the state of iterating through a Dataset.

class Options: Represents options for tf.data.Dataset.

class TFRecordDataset: A Dataset comprising records from one or more TFRecord files.

class TextLineDataset: A Dataset comprising lines from one or more text files.

Functions

get_output_classes(...): Returns the output classes of a Dataset or Iterator elements.

get_output_shapes(...): Returns the output shapes of a Dataset or Iterator elements.

get_output_types(...): Returns the output shapes of a Dataset or Iterator elements.

make_initializable_iterator(...): Creates a tf.compat.v1.data.Iterator for enumerating the elements of a dataset.

make_one_shot_iterator(...): Creates a tf.compat.v1.data.Iterator for enumerating the elements of a dataset.