тензорный поток:: опс:: Восстановить
#include <io_ops.h>
Восстанавливает тензор из файлов контрольных точек.
Краткое содержание
Считывает тензор, хранящийся в одном или нескольких файлах. Если файлов несколько (например, потому что тензор был сохранен как срезы), file_pattern
может содержать подстановочные знаки ( *
и ?
) только в части имени файла, а не в части каталога.
Если file_pattern
соответствует нескольким файлам, preferred_shard
можно использовать, чтобы подсказать, в каком файле, скорее всего, будет найден запрошенный тензор. Эта операция сначала откроет файл с preferred_shard
в списке соответствующих файлов и попытается восстановить тензоры из этого файла. Только если в этом первом файле не найдены некоторые тензоры или тензорные фрагменты, Op открывает все файлы. Установка preferred_shard
в соответствии со значением, переданным в качестве входных данных shard
соответствующей операции Save
, может ускорить восстановление . Этот атрибут влияет только на производительность, но не на правильность. Значение по умолчанию -1 означает, что файлы обрабатываются по порядку.
См. также RestoreSlice
.
Аргументы:
- область: объект области.
- file_pattern: должен содержать один элемент. Паттерн файлов, из которых мы читаем тензор.
- tensor_name: Должен иметь один элемент. Имя восстанавливаемого тензора.
- dt: тип восстанавливаемого тензора.
Необязательные атрибуты (см. Attrs
):
- Preferred_shard: индекс файла, который открывается первым, если несколько файлов соответствуют
file_pattern
.
Возврат:
-
Output
: восстановленный тензор.
Конструкторы и деструкторы | |
---|---|
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt) | |
Restore (const :: tensorflow::Scope & scope, :: tensorflow::Input file_pattern, :: tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs) |
Публичные атрибуты | |
---|---|
operation | |
tensor |
Общественные функции | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Публичные статические функции | |
---|---|
PreferredShard (int64 x) |
Структуры | |
---|---|
tensorflow::ops::Restore::Attrs | Дополнительные установщики атрибутов для Restore . |
Публичные атрибуты
операция
Operation operation
тензор
::tensorflow::Output tensor
Общественные функции
Восстановить
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt )
Восстановить
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs )
узел
::tensorflow::Node * node() const
оператор::tensorflow::Input
operator::tensorflow::Input() const
оператор::tensorflow::Выход
operator::tensorflow::Output() const
Публичные статические функции
PreferredShard
Attrs PreferredShard( int64 x )