docs: public void setInput(int inputNum, INDArray input) {
          if (inputs == null) {
              //May be null after clear()
              inputs = new INDArray[numInputArrays];
          }
          inputs[inputNum] = input;
      }