tf.raw_ops.SaveV2 | TensorFlow v2.16.1 (original) (raw)
tf.raw_ops.SaveV2
Stay organized with collections Save and categorize content based on your preferences.
Saves tensors in V2 checkpoint format.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.raw_ops.SaveV2(
prefix, tensor_names, shape_and_slices, tensors, name=None
)
By default, saves the named tensors in full. If the caller wishes to save specific slices of full tensors, "shape_and_slices" should be non-empty strings and correspondingly well-formed.
Args | |
---|---|
prefix | A Tensor of type string. Must have a single element. The prefix of the V2 checkpoint to which we write the tensors. |
tensor_names | A Tensor of type string. shape {N}. The names of the tensors to be saved. |
shape_and_slices | A Tensor of type string. shape {N}. The slice specs of the tensors to be saved. Empty strings indicate that they are non-partitioned tensors. |
tensors | A list of Tensor objects. N tensors to save. |
name | A name for the operation (optional). |
Returns |
---|
The created Operation. |
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.