tf.raw_ops.TFRecordDataset
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