tf.TFRecordReader
Чтение записей из файла TFRecords.
Наследуется от: ReaderBase
tf.TFRecordReader(
name=None, options=None
)
См. ReaderBase для поддерживаемых методов.
| Аргументы | |
|---|---|
name | Имя операции (необязательно). |
options | Объект TFRecordOptions (необязательно). |
Совместимость с Eager
Читатели не совместимы с eager-вычислением. Вместо этого используйте tf.data для получения данных в вашу модель.
| Атрибуты | |
|---|---|
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
)
Возвращает следующую пару (ключ, значение) записей, полученную читателем.
При необходимости будет извлечен элемент из очереди (например, когда читателю нужно начать чтение из нового файла, так как он завершил чтение предыдущего файла).
| Аргументы | |
|---|---|
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 | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
| Строковый тензор. |
© 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/r1.15/api_docs/python/tf/TFRecordReader