tf.raw_ops.RestoreV2
Восстанавливает тензоры из контрольной точки V2.
tf.raw_ops.RestoreV2(
prefix, tensor_names, shape_and_slices, dtypes, name=None
)
Для обратной совместимости с форматом V1 этот оператор в настоящее время также позволяет выполнять восстановление из контрольной точки V1:
- Сначала этот оператор пытается найти файл индекса V2, на который указывает «prefix», и, если файл найден, считывает его как контрольную точку V2;
- В противном случае используется путь чтения V1. Не рекомендуется полагаться на такое поведение, поскольку возможность переключения на чтение V1 может быть объявлена устаревшей и в конечном итоге удалена.
По умолчанию все указанные тензоры восстанавливаются целиком. Если вызывающей стороне требуется восстановить отдельные срезы сохранённых тензоров, значения «shape_and_slices» должны быть непустыми строками правильного формата.
Вызывающие стороны должны убедиться, что все указанные тензоры действительно сохранены в контрольной точке.
| Аргументы | |
|---|---|
prefix | Tensor типа string. Должен содержать один элемент. Префикс контрольной точки V2. |
tensor_names | Tensor типа string. Форма {N}. Имена восстанавливаемых тензоров. |
shape_and_slices | Tensor типа string. Форма {N}. Спецификации срезов восстанавливаемых тензоров. Пустые строки указывают на то, что тензоры не разбиты на части. |
dtypes | Список tf.DTypes длиной >= 1. Форма {N}. Список ожидаемых типов данных (dtype) тензоров. Они должны совпадать с типами, сохранёнными в контрольной точке. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Список объектов Tensor типа dtypes. |
© 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/api_docs/python/tf/raw_ops/RestoreV2