@Override public void clearVertex() { clear(); epsilon = null; } }
@Override public void setInput(int inputNumber, INDArray input) { if (inputNumber >= getNumInputArrays()) { throw new IllegalArgumentException("Invalid input number"); } inputs[inputNumber] = input; }
@Override public boolean canDoBackward() { if (!isOutputVertex()) { //inputs to frozen layer go unchecked, so could be null if (getLayer() instanceof FrozenLayer) { return true; } else { return super.canDoBackward(); } } for (INDArray input : inputs) { if (input == null) { return false; } } if (!(layer instanceof IOutputLayer)) { if (epsilon == null) { return false; } } return true; } }