Type TFRecordWriter
Namespace tensorflow.data.experimental
Parent PythonObjectContainer
Interfaces ITFRecordWriter
Writes data to a TFRecord file. To write a `dataset` to a single TFRecord file:
To shard a `dataset` across multiple TFRecord files:
Show Example
dataset =... # dataset to be written
writer = tf.data.experimental.TFRecordWriter(PATH)
writer.write(dataset)
Methods
Properties
Public instance methods
object write(Dataset dataset)
Returns a
tf.Operation to write a dataset to a file.
Parameters
-
Datasetdataset - a
tf.data.Datasetwhose elements are to be written to a file
Returns
-
object - A
tf.Operationthat, when run, writes contents of `dataset` to a file.
object write(DatasetV1Adapter dataset)
Returns a
tf.Operation to write a dataset to a file.
Parameters
-
DatasetV1Adapterdataset - a
tf.data.Datasetwhose elements are to be written to a file
Returns
-
object - A
tf.Operationthat, when run, writes contents of `dataset` to a file.