| Package | Description |
|---|---|
| org.tensorflow.op.core |
| Modifier and Type | Method and Description |
|---|---|
static RebatchDataset |
RebatchDataset.create(Scope scope,
Operand<?> inputDataset,
Operand<Long> numReplicas,
List<Class<?>> outputTypes,
List<Shape> outputShapes,
RebatchDataset.Options... options)
Factory method to create a class wrapping a new RebatchDataset operation.
|
Copyright © 2015–2019. All rights reserved.