@Operator(group="data") public final class CacheDataset extends PrimitiveOp implements Operand<java.lang.Object>
A CacheDataset will iterate over the input_dataset, and store tensors. If the cache already exists, the cache will be used. If the cache is inappropriate (e.g. cannot be opened, contains tensors of the wrong shape / size), an error will the returned when used.
operation| Modifier and Type | Method | Description |
|---|---|---|
Output<java.lang.Object> |
asOutput() |
Returns the symbolic handle of a tensor.
|
static CacheDataset |
create(Scope scope,
Operand<?> inputDataset,
Operand<java.lang.String> filename,
java.util.List<java.lang.Class<?>> outputTypes,
java.util.List<Shape> outputShapes) |
Factory method to create a class to wrap a new CacheDataset operation to the graph.
|
Output<?> |
handle() |
clone, finalize, getClass, notify, notifyAll, wait, wait, waitequals, hashCode, toStringpublic static CacheDataset create(Scope scope, Operand<?> inputDataset, Operand<java.lang.String> filename, java.util.List<java.lang.Class<?>> outputTypes, java.util.List<Shape> outputShapes)
scope - current graph scopeinputDataset - filename - A path on the filesystem where we should cache the dataset. Note: this
will be a directory.outputTypes - outputShapes - public Output<?> handle()
public Output<java.lang.Object> asOutput()
OperandInputs 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<java.lang.Object>OperationBuilder.addInput(Output)