docs: public static long vectorsPerSlice(INDArray arr) {
          if (arr.rank() > 2) {
              return ArrayUtil.prodLong(new long[] {arr.size(-1), arr.size(-2)});
          }
  
          return arr.slices();
      }