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 | Tensor типа string. Должен иметь один элемент. Шаблон файлов, из которых мы считываем тензор. |
tensor_name | Tensor типа string. Должен иметь один элемент. Имя тензора, который нужно восстановить. |
dt | tf.DType. Тип тензора, который нужно восстановить. |
preferred_shard | Необязательный int. По умолчанию -1. Индекс файла для первого открытия, если нескольким файлам соответствует file_pattern. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Tensor типа dt. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/raw_ops/Restore
