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.compat.v1.raw_ops.SaveV2

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.