Examples with InputPreProcessor org.deeplearning4j.nn.conf.InputPreProcessor used on opensource projects

Example 6 with InputPreProcessor

use of org.deeplearning4j.nn.conf.InputPreProcessor in project deeplearning4j by deeplearning4j.

the class MultiLayerNetwork method feedForwardMaskArray.

@Override
public Pair<INDArray, MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize) {
    if (maskArray == null) {
        for (int i = 0; i < layers.length; i++) {
            layers[i].feedForwardMaskArray(null, null, minibatchSize);
        }
    } else {
        //Do a forward pass through each preprocessor and layer
        for (int i = 0; i < layers.length; i++) {
            InputPreProcessor preProcessor = getLayerWiseConfigurations().getInputPreProcess(i);
            if (preProcessor != null) {
                Pair<INDArray, MaskState> p = preProcessor.feedForwardMaskArray(maskArray, currentMaskState, minibatchSize);
                if (p != null) {
                    maskArray = p.getFirst();
                    currentMaskState = p.getSecond();
                } else {
                    maskArray = null;
                    currentMaskState = null;
                }
            }
            Pair<INDArray, MaskState> p = layers[i].feedForwardMaskArray(maskArray, currentMaskState, minibatchSize);
            if (p != null) {
                maskArray = p.getFirst();
                currentMaskState = p.getSecond();
            } else {
                maskArray = null;
                currentMaskState = null;
            }
        }
    }
    return new Pair<>(maskArray, currentMaskState);
}