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 )
© 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/versions/r2.9/api_docs/cc/class/tensorflow/ops/restore