org.tensorflow.op.train

Class Save

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.train.Save

All Implemented Interfaces:

Op

@Operator(group="train")
public final class Save
extends PrimitiveOp

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.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static Save	create(Scope scope, Operand<String> prefix, Operand<String> tensorNames, Operand<String> shapeAndSlices, Iterable<Operand<?>> tensors) Factory method to create a class wrapping a new Save operation.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static Save create(Scope scope,
                          Operand<String> prefix,
                          Operand<String> tensorNames,
                          Operand<String> 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