@Operator(group="train") public final class SaveSlices extends PrimitiveOp
This is like `Save` except that tensors can be listed in the saved file as being a slice of a larger tensor. `shapes_and_slices` specifies the shape of the larger tensor and the slice that this tensor covers. `shapes_and_slices` must have as many elements as `tensor_names`.
Elements of the `shapes_and_slices` input must either be:
operation
Modifier and Type | Method and Description |
---|---|
static SaveSlices |
create(Scope scope,
Operand<String> filename,
Operand<String> tensorNames,
Operand<String> shapesAndSlices,
Iterable<Operand<?>> data)
Factory method to create a class wrapping a new SaveSlices operation.
|
equals, hashCode, op, toString
public static SaveSlices create(Scope scope, Operand<String> filename, Operand<String> tensorNames, Operand<String> shapesAndSlices, Iterable<Operand<?>> data)
scope
- current scopefilename
- Must have a single element. The name of the file to which we write the
tensor.tensorNames
- Shape `[N]`. The names of the tensors to be saved.shapesAndSlices
- Shape `[N]`. The shapes and slice specifications to use when
saving the tensors.data
- `N` tensors to save.Copyright © 2022. All rights reserved.