code: @Override
      public void rnnSetPreviousState(Map<String, INDArray> stateMap) {
          this.stateMap.clear();
          this.stateMap.putAll(stateMap);
      }
docs: Set the state map. Values set using this method will be used
  in next call to rnnTimeStep()