Method Detail

• getInputBatches

  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). In these "can't batch" cases, multiple input batches will be returned, to be processed

  Returns:

  List of pairs of input arrays and input mask arrays. Input mask arrays may be null.

• addInput

  void addInput(INDArray... input)

• addInput

  void addInput(INDArray[] input,
                INDArray[] inputMasks)

• setOutputBatches

  void setOutputBatches(java.util.List<INDArray[]> output)

• setOutputException

  void setOutputException(java.lang.Exception e)

• addObserver

  void addObserver(java.util.Observer observer)

• getOutput

  INDArray[] getOutput()