tensorflow::ops::RestoreV2
#include <io_ops.h>
Восстанавливает тензоры из контрольной точки формата V2.
Краткое описание
Для обеспечения обратной совместимости с форматом V1, данный оператор в настоящее время позволяет восстанавливать данные из контрольной точки формата V1:
- Оператор сначала пытается найти файл индексов V2, указанный префиксом "prefix", и если находит, то считывает его как контрольную точку V2;
- В противном случае используется путь чтения V1. Не рекомендуется полагаться на это поведение, так как возможность падения к чтению V1 может быть устарела и в конечном итоге удалена.
По умолчанию восстанавливаются все тензоры с указанными именами. Если вызывающий код желает восстановить определённые срезы сохранённых тензоров, "shape_and_slices" должно содержать непустые строки соответствующего формата.
Вызывающий код должен убедиться, что все указанные тензоры действительно хранятся в контрольной точке.
Аргументы:
- scope: Объект Scope
- prefix: Должен содержать единственный элемент. Префикс контрольной точки V2.
- tensor_names: массив {N}. Имена тензоров, которые нужно восстановить.
- shape_and_slices: массив {N}. Спецификации срезов тензоров, которые нужно восстановить. Пустые строки указывают на непропорциональные тензоры.
- dtypes: массив {N}. Список ожидаемых типов данных для тензоров. Должен совпадать с типами данных, сохранёнными в контрольной точке.
Возвращает:
-
OutputList: массив {N}. Восстановленные тензоры, формы которых считываются из контрольной точки напрямую.
| Конструкторы и деструкторы | |
|---|---|
RestoreV2(const ::tensorflow::Scope & scope, ::tensorflow::Input prefix, ::tensorflow::Input tensor_names, ::tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes) |
| Публичные атрибуты | |
|---|---|
operation | |
tensors | |
| Публичные функции | |
|---|---|
operator[](size_t index) const | |
Публичные атрибуты
operation
Operation operation
tensors
::tensorflow::OutputList tensors
Публичные функции
RestoreV2
RestoreV2( const ::tensorflow::Scope & scope, ::tensorflow::Input prefix, ::tensorflow::Input tensor_names, ::tensorflow::Input shape_and_slices, const DataTypeSlice & dtypes )
operator[]
::tensorflow::Output operator[]( size_t index ) const
© 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-v2