How to convert NDArray to double[][]

[jamalromero]

Hi,
I am looking for an efficient way to convert an NDArray with shape nxm to an array of double like 'double[][]'. Is there a builtin method in NDArray or NDManager or in the package to achieve that. I know there's .toDoubleArray() but it "flattens" the NDArray to 1D. Example, I want to convert:

ND: (4, 4) cpu() float64
[[4., 5., 6., 7.],
 [3., 4., 5., 6.],
 [2., 3., 4., 5.],
 [1., 2., 3., 4.],
]

to

double[][] arr = new double[][]
{{4., 5., 6., 7.},
 {3., 4., 5., 6.},
 {2., 3., 4., 5.},
 {1., 2., 3., 4.},
}

My method is this:

double[][] convert(NDArray ndArray) {
  List<double[]> list = new ArrayList<>();
  for (int i = 0; i < ndArray.getShape().get(0); i++) {
    list.add(ndArray.get(i).toDoubleArray());
  }
  double[][] array = new double[list.size()][];
  for (int i = 0; i < list.size(); i++) {
    array[i] = list.get(i);
  }
  return array;
}

But I am wondering if there's a builtin and efficient way to achieve this using the same library.
Thanks