docs: public static long[] ind2sub(INDArray arr, long index) {
          if (arr.rank() == 1)
              return new long[]{(int) index};
          return ind2sub(arr.shape(), index, ArrayUtil.prodLong(arr.shape()));
      }