public abstract class BaseParallelDataSetIterator extends Object implements org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIterator
| Modifier and Type | Field and Description |
|---|---|
protected AtomicBoolean |
allDepleted |
protected AtomicLong |
counter |
protected org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling |
inequalityHandling |
protected int |
numProducers |
protected ThreadLocal<Integer> |
producerAffinity |
protected MultiBoolean |
resetTracker |
protected MultiBoolean |
states |
| Modifier | Constructor and Description |
|---|---|
protected |
BaseParallelDataSetIterator(int numProducers) |
| Modifier and Type | Method and Description |
|---|---|
boolean |
asyncSupported() |
void |
attachThread(int producer) |
int |
batch() |
int |
cursor() |
protected int |
getCurrentProducerIndex() |
List<String> |
getLabels() |
org.nd4j.linalg.dataset.api.DataSetPreProcessor |
getPreProcessor() |
boolean |
hasNext() |
boolean |
hasNextFor() |
abstract boolean |
hasNextFor(int consumer) |
int |
inputColumns() |
org.nd4j.linalg.dataset.DataSet |
next() |
org.nd4j.linalg.dataset.DataSet |
next(int num) |
org.nd4j.linalg.dataset.DataSet |
nextFor() |
abstract org.nd4j.linalg.dataset.DataSet |
nextFor(int consumer) |
int |
numExamples() |
void |
remove() |
void |
reset() |
protected abstract void |
reset(int consumer) |
boolean |
resetSupported() |
void |
setPreProcessor(org.nd4j.linalg.dataset.api.DataSetPreProcessor preProcessor) |
protected void |
stepForward() |
int |
totalExamples() |
int |
totalOutcomes() |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitforEachRemainingprotected AtomicLong counter
protected org.nd4j.linalg.dataset.api.iterator.enums.InequalityHandling inequalityHandling
protected int numProducers
protected AtomicBoolean allDepleted
protected MultiBoolean states
protected MultiBoolean resetTracker
protected ThreadLocal<Integer> producerAffinity
protected BaseParallelDataSetIterator(int numProducers)
public boolean hasNext()
public org.nd4j.linalg.dataset.DataSet next()
protected int getCurrentProducerIndex()
protected void stepForward()
public void reset()
reset in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic void attachThread(int producer)
attachThread in interface org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIteratorpublic boolean hasNextFor()
hasNextFor in interface org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIteratorpublic org.nd4j.linalg.dataset.DataSet nextFor()
nextFor in interface org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIteratorpublic abstract boolean hasNextFor(int consumer)
hasNextFor in interface org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIteratorpublic abstract org.nd4j.linalg.dataset.DataSet nextFor(int consumer)
nextFor in interface org.nd4j.linalg.dataset.api.iterator.ParallelDataSetIteratorprotected abstract void reset(int consumer)
public int totalExamples()
totalExamples in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic int totalOutcomes()
totalOutcomes in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic boolean resetSupported()
resetSupported in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic boolean asyncSupported()
asyncSupported in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic int batch()
batch in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic int cursor()
cursor in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic int numExamples()
numExamples in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic org.nd4j.linalg.dataset.DataSet next(int num)
next in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic int inputColumns()
inputColumns in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic void setPreProcessor(org.nd4j.linalg.dataset.api.DataSetPreProcessor preProcessor)
setPreProcessor in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic org.nd4j.linalg.dataset.api.DataSetPreProcessor getPreProcessor()
getPreProcessor in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorpublic List<String> getLabels()
getLabels in interface org.nd4j.linalg.dataset.api.iterator.DataSetIteratorCopyright © 2017. All rights reserved.