тензорный поток:: опс:: Восстановить
#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 )