docs: public INDArray getProbabilityMatrix(INDArray networkOutput, int example, int classNumber){
          //Input format: [minibatch, 5B+C, H, W], with order [x,y,w,h,c]
          //Therefore: probabilities for class I is at depths 5B + classNumber
  
          val bbs = layerConf().getBoundingBoxes().size(0);
          INDArray conf = networkOutput.get(point(example), point(5*bbs + classNumber), all(), all());
          return conf;
      }