public class BatchedInferenceObservable extends BasicInferenceObservable implements InferenceObservable
exception
Constructor and Description |
---|
BatchedInferenceObservable() |
Modifier and Type | Method and Description |
---|---|
void |
addInput(INDArray[] input,
INDArray[] inputMasks) |
int |
getCounter() |
java.util.List<Pair<INDArray[],INDArray[]>> |
getInputBatches()
Get input batches - and their associated input mask arrays, if any
Note that usually the returned list will be of size 1 - however, in the batched case, not all inputs can actually be batched (variable size inputs to fully convolutional net, for example). |
INDArray[] |
getOutput() |
protected java.util.List<INDArray[]> |
getOutputs()
PLEASE NOTE: This method is for tests only
|
boolean |
isLocked() |
protected void |
setCounter(int value) |
void |
setOutputBatches(java.util.List<INDArray[]> output) |
void |
setPosition(int pos) |
addInput, checkOutputException, setOutputException
addObserver, clearChanged, countObservers, deleteObserver, deleteObservers, hasChanged, notifyObservers, notifyObservers, setChanged
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
addInput, addObserver, setOutputException
public void addInput(INDArray[] input, INDArray[] inputMasks)
addInput
in interface InferenceObservable
addInput
in class BasicInferenceObservable
public java.util.List<Pair<INDArray[],INDArray[]>> getInputBatches()
InferenceObservable
getInputBatches
in interface InferenceObservable
getInputBatches
in class BasicInferenceObservable
public void setOutputBatches(java.util.List<INDArray[]> output)
setOutputBatches
in interface InferenceObservable
setOutputBatches
in class BasicInferenceObservable
protected java.util.List<INDArray[]> getOutputs()
protected void setCounter(int value)
public void setPosition(int pos)
public int getCounter()
public boolean isLocked()
public INDArray[] getOutput()
getOutput
in interface InferenceObservable
getOutput
in class BasicInferenceObservable