@Operator(group="train") public final class Save extends PrimitiveOp
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.
operation
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.
|
equals, hashCode, op, toString
public static Save create(Scope scope, Operand<String> prefix, Operand<String> tensorNames, Operand<String> shapeAndSlices, Iterable<Operand<?>> tensors)
scope
- current scopeprefix
- 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.Copyright © 2022. All rights reserved.