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 | Очередь или изменяемый строковый тензор, представляющий дескриптор очереди со строковыми элементами работы. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
| Кортеж тензоров (ключ, значение). | |
key | Строковый скалярный тензор. |
value | Строковый скалярный тензор. |
read_up_to
read_up_to(
queue, num_records, name=None
)
Возвращает до num_records пар (ключ, значение) записей.
Если необходимо, декьюит единицу работы из очереди (например, когда чтение должно начаться с нового файла, так как предыдущий закончился). Может вернуть меньше 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/api_docs/python/tf/compat/v1/TFRecordReader