code: @Override
      public void onNDArrayMessage(NDArrayMessage message) {
          INDArray arr = message.getArr();
          //of note for ndarrays
          int[] dimensions = message.getDimensions();
          boolean whole = dimensions.length == 1 && dimensions[0] == -1;
  
          if (!whole)
              onNDArrayPartial(arr, message.getIndex(), dimensions);
          else
              onNDArray(arr);
      }
docs: A listener for ndarray message
  
  @param message the message for the callback