docs: public static INDArrayIndex[] indexesFor(long... shape) {
          INDArrayIndex[] ret = new INDArrayIndex[shape.length];
          for (int i = 0; i < shape.length; i++) {
              ret[i] = NDArrayIndex.point(shape[i]);
          }
  
          return ret;
      }