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