public class
TakeDataset
Public Constructors
Inherited Methods
final Dataset |
batch(long batchSize, boolean dropLastBatch)
Groups elements of this dataset into batches.
|
final Dataset |
batch(long batchSize)
Groups elements of this dataset into batches.
|
static Dataset |
fromTensorSlices(Ops tf, List<Operand<?>> tensors, List<Class<? extends TType>> outputTypes)
Creates an in-memory `Dataset` whose elements are slices of the given tensors.
|
Ops | |
List<Shape> |
getOutputShapes()
Get a list of shapes for each component of this dataset.
|
List<Class<? extends TType>> |
getOutputTypes()
Get a list of output types for each component of this dataset.
|
Operand<?> |
getVariant()
Get the variant tensor representing this dataset.
|
Iterator<List<Operand<?>>> |
iterator()
Creates an iterator which iterates through all batches of this Dataset in an eager fashion.
|
DatasetIterator |
makeInitializeableIterator()
Creates a `DatasetIterator` that can be used to iterate over elements of this dataset.
|
DatasetIterator |
makeOneShotIterator()
Creates a `DatasetIterator` that can be used to iterate over elements of this dataset.
|
Dataset | |
Dataset |
mapAllComponents(Function<Operand<?>, Operand<?>> mapper)
Returns a new Dataset which maps a function across all elements from this dataset, on all
components of each element.
|
Dataset |
mapOneComponent(int index, Function<Operand<?>, Operand<?>> mapper)
Returns a new Dataset which maps a function across all elements from this dataset, on a single
component of each element.
|
final Dataset |
skip(long count)
Returns a new `Dataset` which skips `count` initial elements from this dataset
|
final Dataset |
take(long count)
Returns a new `Dataset` with only the first `count` elements from this dataset.
|
static Dataset |
textLineDataset(Ops tf, String filename, String compressionType, long bufferSize)
|
static Dataset |
tfRecordDataset(Ops tf, String filename, String compressionType, long bufferSize)
|
String |
toString()
|
boolean |
equals(Object arg0)
|
final Class<?> |
getClass()
|
int |
hashCode()
|
final void |
notify()
|
final void |
notifyAll()
|
String |
toString()
|
final void |
wait(long arg0, int arg1)
|
final void |
wait(long arg0)
|
final void |
wait()
|