Writes the given dataset to the given file using the TFRecord format.
tf.raw_ops.DatasetToTFRecord(
input_dataset, filename, compression_type, name=None
)
| Args | |
|---|---|
input_dataset | A Tensor of type variant. A variant tensor representing the dataset to write. |
filename | A Tensor of type string. A scalar string tensor representing the filename to use. |
compression_type | A Tensor of type string. A scalar string tensor containing either (i) the empty string (no compression), (ii) "ZLIB", or (iii) "GZIP". |
name | A name for the operation (optional). |
| Returns | |
|---|---|
| The created Operation. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/DatasetToTFRecord