code: @Override
      public synchronized void onNDArray(INDArray arr) {
          if (shape == null)
              updater.update(arr.reshape(1, arr.length()), updater.ndArrayHolder().get());
          else
              updater.update(arr, updater.ndArrayHolder().get());
      }
docs: Setup an ndarray
  
  @param arr