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.4/api_docs/python/tf/raw_ops/RestoreV2