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 может ускорить Restore. Этот атрибут влияет только на производительность, а не на правильность. Значение по умолчанию -1 означает, что файлы обрабатываются в порядке.
См. также RestoreSlice.
| Аргументы | |
|---|---|
file_pattern | Список Tensor типа string. Должен иметь один элемент. Шаблон файлов, из которых считывается тензор. |
tensor_name | Список Tensor типа string. Должен иметь один элемент. Имя тензора, который необходимо восстановить. |
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/versions/r2.9/api_docs/python/tf/raw_ops/Restore
