Modifier and Type | Class and Description |
---|---|
static class |
PaddedBatchDataset.Options
Optional attributes for
PaddedBatchDataset |
operation
Modifier and Type | Method and Description |
---|---|
Output<Object> |
asOutput()
Returns the symbolic handle of a tensor.
|
static PaddedBatchDataset |
create(Scope scope,
Operand<?> inputDataset,
Operand<Long> batchSize,
Iterable<Operand<Long>> paddedShapes,
Iterable<Operand<?>> paddingValues,
Operand<Boolean> dropRemainder,
List<Shape> outputShapes,
PaddedBatchDataset.Options... options)
Factory method to create a class wrapping a new PaddedBatchDataset operation.
|
Output<?> |
handle() |
static PaddedBatchDataset.Options |
parallelCopy(Boolean parallelCopy) |
equals, hashCode, op, toString
public static PaddedBatchDataset create(Scope scope, Operand<?> inputDataset, Operand<Long> batchSize, Iterable<Operand<Long>> paddedShapes, Iterable<Operand<?>> paddingValues, Operand<Boolean> dropRemainder, List<Shape> outputShapes, PaddedBatchDataset.Options... options)
scope
- current scopeinputDataset
- batchSize
- A scalar representing the number of elements to accumulate in a
batch.paddedShapes
- A list of int64 tensors representing the desired padded shapes
of the corresponding output components. These shapes may be partially
specified, using `-1` to indicate that a particular dimension should be
padded to the maximum size of all batch elements.paddingValues
- A list of scalars containing the padding value to use for
each of the outputs.dropRemainder
- A scalar representing whether the last batch should be dropped in case its size
is smaller than desired.outputShapes
- options
- carries optional attributes valuespublic static PaddedBatchDataset.Options parallelCopy(Boolean parallelCopy)
parallelCopy
- public Output<?> handle()
public Output<Object> 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<Object>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.