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