tf.contrib.data.TFRecordWriter

Writes data to a TFRecord file.

Inherits From: TFRecordWriter

tf.contrib.data.TFRecordWriter(
    filename, compression_type=None
)

Methods

write

View source

write(
    dataset
)

Returns a tf.Operation to write a dataset to a file.

Args
dataset	a tf.data.Dataset whose elements are to be written to a file

Returns
A tf.Operation that, when run, writes contents of dataset to a file.