org.tensorflow.op.core

Class SaveV2

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.SaveV2

All Implemented Interfaces:

Op

@Operator
public final class SaveV2
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 SaveV2	create(Scope scope, Operand<String> prefix, Operand<String> tensorNames, Operand<String> shapeAndSlices, Iterable<Operand<?>> tensors) Factory method to create a class to wrap a new SaveV2 operation to the graph.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

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

Factory method to create a class to wrap a new SaveV2 operation to the graph.

Parameters:

scope - current graph 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 SaveV2