tf.compat.v1.TFRecordReader
Чтение записей из файла TFRecords.
Наследуется от: ReaderBase
tf.compat.v1.TFRecordReader(
name=None, options=None
)
См. ReaderBase для поддерживаемых методов.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
options | Объект TFRecordOptions (необязательно). |
| Атрибуты | |
|---|---|
reader_ref | Операция, реализующая чтение. |
supports_serialize | Может ли реализация Reader сохранять своё состояние. |
Методы
num_records_produced
num_records_produced(
name=None
)
Возвращает количество записей, обработанных этим читателем.
Это соответствует количеству успешных вызовов Read.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Int64 тензор. |
num_work_units_completed
num_work_units_completed(
name=None
)
Возвращает количество завершенных единиц работы этим читателем.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Int64 тензор. |
read
read(
queue, name=None
)
Возвращает следующую пару (ключ, значение) из записей читателя.
Если необходимо, будет извлечена единица работы из очереди (например, когда Reader должен начать чтение из нового файла, так как предыдущий файл был обработан).
| Аргументы | |
|---|---|
queue | Очередь или изменяемый строковый тензор, представляющий дескриптор очереди со строковыми элементами работы. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Кортеж тензоров (ключ, значение). | |
key | Строковый скалярный тензор. |
value | Строковый скалярный тензор. |
read_up_to
read_up_to(
queue, num_records, name=None
)
Возвращает до num_records пар (ключ, значение), считанных читателем.
Если необходимо, будет извлечена единица работы из очереди (например, когда Reader должен начать чтение из нового файла, так как предыдущий был обработан). Может вернуть меньше num_records даже перед последней партией.
| Аргументы | |
|---|---|
queue | Очередь или изменяемый строковый тензор, представляющий дескриптор очереди со строковыми элементами работы. |
num_records | Количество записей для чтения. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Кортеж тензоров (ключи, значения). | |
keys | Строковый тензор 1-D. |
values | Строковый тензор 1-D. |
reset
reset(
name=None
)
Восстановление состояния читателя в начальное чистое состояние.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
restore_state
restore_state(
state, name=None
)
Восстановление состояния читателя из ранее сохранённого состояния.
Не все Читатели поддерживают восстановление, поэтому это может привести к ошибке Unimplemented.
| Аргументы | |
|---|---|
state | Строковый тензор. Результат метода SerializeState читателя с соответствующим типом. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Созданная операция. |
serialize_state
serialize_state(
name=None
)
Генерирует строковый тензор, кодирующий состояние читателя.
Не все Читатели поддерживают сериализацию, поэтому это может привести к ошибке Unimplemented.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Строковый тензор. |
совместимость с eager режимом
Читатели не совместимы с eager выполнением. Используйте tf.data для получения данных в вашей модели.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/TFRecordReader