public interface InferenceObservable
Modifier and Type | Method and Description |
---|---|
void |
addInput(INDArray... input) |
void |
addInput(INDArray[] input,
INDArray[] inputMasks) |
void |
addObserver(java.util.Observer observer) |
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() |
void |
setOutputBatches(java.util.List<INDArray[]> output) |
void |
setOutputException(java.lang.Exception e) |
java.util.List<Pair<INDArray[],INDArray[]>> getInputBatches()
void addInput(INDArray... input)
void setOutputBatches(java.util.List<INDArray[]> output)
void setOutputException(java.lang.Exception e)
void addObserver(java.util.Observer observer)
INDArray[] getOutput()