Creates a dataset that emits the records from one or more TFRecord files.
tf.raw_ops.TFRecordDataset( filenames, compression_type, buffer_size, name=None )
Args | |
---|---|
filenames | A Tensor of type string . A scalar or vector containing the name(s) of the file(s) to be read. |
compression_type | A Tensor of type string . A scalar containing either (i) the empty string (no compression), (ii) "ZLIB", or (iii) "GZIP". |
buffer_size | A Tensor of type int64 . A scalar representing the number of bytes to buffer. A value of 0 means no buffering will be performed. |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type variant . |
© 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/TFRecordDataset