Missed TensorFlow World? Check out the recap. Learn more

tf.data.experimental.TFRecordWriter

TensorFlow 2.0 version View source on GitHub

Class TFRecordWriter

Writes data to a TFRecord file.

Aliases:

  • Class tf.compat.v1.data.experimental.TFRecordWriter
  • Class tf.compat.v2.data.experimental.TFRecordWriter

__init__

View source

__init__(
    filename,
    compression_type=None
)

Methods

write

View source

write(dataset)

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

Args:

Returns:

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