tf.compat.v1.TFRecordReader
Считыватель, который выдаёт записи из файла TFRecords.
Наследуется от: ReaderBase
tf.compat.v1.TFRecordReader(
name=None, options=None
)
Список поддерживаемых методов см. в ReaderBase.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
options | Объект TFRecordOptions (необязательно). |
| Атрибуты | |
|---|---|
reader_ref | Операция, реализующая считыватель. |
supports_serialize | Может ли реализация считывателя сериализовать своё состояние. |
Методы
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
)
Возвращает следующую пару (ключ, значение), выданную считывателем.
При необходимости извлекает рабочую единицу из очереди (например, когда считывателю нужно начать чтение нового файла, поскольку обработка предыдущего файла завершена).
| Аргументы | |
|---|---|
queue | Queue или изменяемый строковый тензор, представляющий дескриптор Queue с рабочими элементами типа string. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
| Кортеж тензоров (ключ, значение). | |
key | Скалярный строковый тензор. |
value | Скалярный строковый тензор. |
read_up_to
read_up_to(
queue, num_records, name=None
)
Возвращает до num_records пар (ключ, значение), выданных считывателем.
При необходимости извлекает рабочую единицу из очереди (например, когда считывателю нужно начать чтение нового файла, поскольку обработка предыдущего файла завершена). Даже до последней порции он может вернуть меньше записей, чем указано в num_records.
| Аргументы | |
|---|---|
queue | Queue или изменяемый строковый тензор, представляющий дескриптор Queue с рабочими элементами типа string. |
num_records | Количество записей для чтения. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
| Кортеж тензоров (ключи, значения). | |
keys | Одномерный строковый тензор. |
values | Одномерный строковый тензор. |
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 execution
Считыватели несовместимы с eager execution. Вместо этого используйте 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/api_docs/python/tf/compat/v1/TFRecordReader