tf.raw_ops.Restore
Восстанавливает тензор из файлов контрольной точки.
tf.raw_ops.Restore(
file_pattern, tensor_name, dt, preferred_shard=-1, name=None
)
Используется в ноутбуках
| Используется в руководствах |
|---|
Читает тензор, хранящийся в одном или нескольких файлах. Если файлов несколько (например, если тензор был сохранен в виде фрагментов), file_pattern может содержать символы подстановки (* и ?) только в части имени файла, но не в части каталога.
Если file_pattern соответствует нескольким файлам, можно использовать preferred_shard, чтобы указать, в каком файле, вероятнее всего, находится запрошенный тензор. Эта операция сначала откроет файл с индексом preferred_shard в списке соответствующих файлов и попытается восстановить тензоры из этого файла. Только если некоторые тензоры или фрагменты тензоров не найдены в первом файле, операция откроет все файлы. Установка значения preferred_shard, совпадающего со значением, переданным в качестве входных данных shard для соответствующей операции Save, может ускорить восстановление. Этот атрибут влияет только на производительность, но не на корректность. Значение по умолчанию -1 означает, что файлы обрабатываются по порядку.
См. также RestoreSlice.
| Аргументы | |
|---|---|
file_pattern | string типа Tensor. Должен содержать один элемент. Шаблон файлов, из которых считывается тензор. |
tensor_name | string типа Tensor. Должен содержать один элемент. Имя восстанавливаемого тензора. |
dt | tf.DType. Тип восстанавливаемого тензора. |
preferred_shard | Необязательный int. По умолчанию — -1. Индекс файла, который нужно открыть первым, если file_pattern соответствует нескольким файлам. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Tensor типа dt. |
© 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/Restore
