tensorflow::ops::Restore
#include <io_ops.h>
Восстанавливает тензор из файлов контрольных точек.
Краткое описание
Читает тензор, сохранённый в одном или нескольких файлах. Если файлов несколько (например, потому что тензор был сохранён как куски), имя файла в file_pattern может содержать символы подстановки (* и ?) только в части имени файла, а не в части каталога.
Если шаблон file_pattern соответствует нескольким файлам, preferred_shard можно использовать для указания, в каком файле, скорее всего, находится запрашиваемый тензор. Этот оператор сначала откроет файл с индексом preferred_shard в списке соответствующих файлов и попытается восстановить тензоры из этого файла. Только если некоторые тензоры или части тензоров не найдены в этом первом файле, оператор откроет все файлы. Установка preferred_shard для соответствия значению, переданному как вход shard оператора, соответствующего Save оператору, может ускорить Restore. Этот атрибут влияет только на производительность, а не на корректность. Значение по умолчанию -1 означает, что файлы обрабатываются в порядке.
См. также RestoreSlice.
Аргументы:
- scope: Объект Scope
- 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 operation
tensor
::tensorflow::Output tensor
Общедоступные функции
Restore
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt )
Restore
Restore( const ::tensorflow::Scope & scope, ::tensorflow::Input file_pattern, ::tensorflow::Input tensor_name, DataType dt, const Restore::Attrs & attrs )
node
::tensorflow::Node * node() const
operator::tensorflow::Input
operator::tensorflow::Input() const
operator::tensorflow::Output
operator::tensorflow::Output() const
Общедоступные статические функции
PreferredShard
Attrs PreferredShard( int64 x )
© 2020 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/versions/r2.4/api_docs/cc/class/tensorflow/ops/restore