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 | A Tensor типа string. Должен иметь один элемент. Шаблон файлов, из которых считывается тензор. |
tensor_name | A Tensor типа string. Должен иметь один элемент. Имя восстанавливаемого тензора. |
dt | A tf.DType. Тип восстанавливаемого тензора. |
preferred_shard | Необязательный int. По умолчанию -1. Индекс файла, который нужно открыть в первую очередь, если нескольким файлам соответствует file_pattern. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
A 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
