docs: public static long matricesPerSlice(INDArray arr) {
          if (arr.rank() == 3) {
              return 1;
          } else if (arr.rank() > 3) {
              int ret = 1;
              for (int i = 1; i < arr.rank() - 2; i++) {
                  ret *= arr.size(i);
              }
              return ret;
          }
          return arr.size(-2);
      }