T
- data type for tensor()
output@Operator(group="train") public final class RestoreSlice<T> extends PrimitiveOp implements Operand<T>
This is like `Restore` except that restored tensor can be listed as filling only a slice of a larger tensor. `shape_and_slice` specifies the shape of the larger tensor and the slice that the restored tensor covers.
The `shape_and_slice` input has the same format as the elements of the `shapes_and_slices` input of the `SaveSlices` op.
Modifier and Type | Class and Description |
---|---|
static class |
RestoreSlice.Options
Optional attributes for
RestoreSlice |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> RestoreSlice<T> |
create(Scope scope,
Operand<String> filePattern,
Operand<String> tensorName,
Operand<String> shapeAndSlice,
Class<T> dt,
RestoreSlice.Options... options)
Factory method to create a class wrapping a new RestoreSlice operation.
|
static RestoreSlice.Options |
preferredShard(Long preferredShard) |
Output<T> |
tensor()
The restored tensor.
|
equals, hashCode, op, toString
public static <T> RestoreSlice<T> create(Scope scope, Operand<String> filePattern, Operand<String> tensorName, Operand<String> shapeAndSlice, Class<T> dt, RestoreSlice.Options... options)
scope
- current scopefilePattern
- Must have a single element. The pattern of the files from
which we read the tensor.tensorName
- Must have a single element. The name of the tensor to be
restored.shapeAndSlice
- Scalar. The shapes and slice specifications to use when
restoring a tensors.dt
- The type of the tensor to be restored.options
- carries optional attributes valuespublic static RestoreSlice.Options preferredShard(Long preferredShard)
preferredShard
- Index of file to open first if multiple files match
`file_pattern`. See the documentation for `Restore`.public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.