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. shape {N}. Имена тензоров, которые необходимо сохранить. |
shape_and_slices | A Tensor типа string. shape {N}. Спецификации фрагментов тензоров, которые необходимо сохранить. Пустые строки указывают, что это неразделенные тензоры. |
tensors | Список объектов Tensor. Сохраняемые тензоры N. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
| Созданная операция. |
© 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/SaveV2
