Save

public final class Save

Saves tensors in V2 checkpoint format.

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.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static Save
create(Scope scope, Operand<TString> prefix, Operand<TString> tensorNames, Operand<TString> shapeAndSlices, Iterable<Operand<?>> tensors)
Factory method to create a class wrapping a new Save operation.

Inherited Methods

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "SaveV2"

Public Methods

public static Save create (Scope scope, Operand<TString> prefix, Operand<TString> tensorNames, Operand<TString> shapeAndSlices, Iterable<Operand<?>> tensors)

Factory method to create a class wrapping a new Save operation.

Parameters
scope current scope
prefix Must have a single element. The prefix of the V2 checkpoint to which we write the tensors.
tensorNames shape {N}. The names of the tensors to be saved.
shapeAndSlices shape {N}. The slice specs of the tensors to be saved. Empty strings indicate that they are non-partitioned tensors.
tensors `N` tensors to save.
Returns
  • a new instance of Save