tf.raw_ops.RestoreV2
Восстанавливает тензоры из контрольной точки V2.
tf.raw_ops.RestoreV2(
prefix, tensor_names, shape_and_slices, dtypes, name=None
)
Для обеспечения обратной совместимости с форматом V1 этот оператор в настоящее время также позволяет восстанавливать данные из контрольной точки V1:
- Этот оператор сначала пытается найти файл индексов V2, на который указывает «prefix», и, если он найден, продолжает чтение как контрольная точка V2;
- В противном случае вызывается путь чтения V1. Не рекомендуется полагаться на это поведение, так как возможность отката к чтению V1 может быть устарела и в конечном итоге удалена.
По умолчанию восстанавливаются все указанные тензоры полностью. Если вызывающая сторона хочет восстановить определенные фрагменты хранимых тензоров, «shape_and_slices» должны быть непустыми строками и соответствующим образом обработаны.
Вызывающая сторона должна гарантировать, что все указанные тензоры действительно хранятся в контрольной точке.
| Аргументы | |
|---|---|
prefix | A Tensor типа string. Должен иметь один элемент. Префикс контрольной точки V2. |
tensor_names | A Tensor типа string. shape {N}. Имена тензоров, которые должны быть восстановлены. |
shape_and_slices | A Tensor типа string. shape {N}. Спецификации фрагментов тензоров, которые должны быть восстановлены. Пустые строки указывают, что это неразделенные тензоры. |
dtypes | Список tf.DTypes длиной >= 1. shape {N}. Список ожидаемых типов данных для тензоров. Должен совпадать с теми, что хранятся в контрольной точке. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Список объектов Tensor типа dtypes. |
© 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/RestoreV2