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 | tf.DType. Тип тензора, который нужно восстановить. |
preferred_shard | Необязательный int. По умолчанию -1. Индекс файла, который нужно открыть в первую очередь, если совпадений с file_pattern несколько. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A 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.4/api_docs/python/tf/raw_ops/Restore
