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. |
© 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/RestoreV2
