java - DMatrixRMaj : get data in matrix form

Question

当我在 ejml(实矩阵的标准格式)中初始化一个新的 DMatrixRMaj 时，它可以在内部存储一个 double[][] 矩阵。示例

double[][] a = new double[][];
//init a
DMatrixRMaj d = new DMatrixRMaj(a);
//math operations on d

现在，经过必要的计算后，我如何才能得到 d 的 double[][] 形式？使用 d.getData() 我只能获取行形式。我也试过用 SimpleMatrix 包装，或从 double 创建 SimpleMatrix，但我没有找到任何方法(或矩阵格式)来检索 double !

你知道我该怎么做吗？或者您可以提出一个无需编写个性化函数的直接解决方法吗？

Answer 1

我不是很熟悉这个库，但是考虑到矩阵是使用 data[ y*numCols + x ] = data[y][x] 格式存储的，你可以编写自己的函数来使用相同的格式检索数据。

示例:.

import org.ejml.data.DMatrixRMaj;

public class Main {

    public static void main(String[] args) throws Exception {
        double [] [] doubles = new double[2][2];
        for (int i = 0; i < 2; i++) {
            for (int j = 0; j < 2; j++) {
                doubles[i][j] = 0.1;
            }
        }

        //init a
        DMatrixRMaj d = new DMatrixRMaj(doubles);

        double [] [] doublesAfter = getInitMatrix(d);

        System.out.println("Initial matrix: ");
        printMatrix(doubles);

        System.out.println("Same matrix after DMatrixRMaj: ");
        printMatrix(doublesAfter);
    }

    private static double [][] getInitMatrix(DMatrixRMaj dMatrixRMaj) {
        double [] [] doubles = new double[dMatrixRMaj.getNumRows()][dMatrixRMaj.getNumCols()];
        for (int x = 0; x < dMatrixRMaj.getNumRows(); x++) {
            for (int y = 0; y <dMatrixRMaj.getNumCols(); y++) {
                doubles[x][y] = dMatrixRMaj.getData()[y*dMatrixRMaj.getNumCols() + x];
            }
        }
        return doubles;
    }

    private static void printMatrix(double [][] matrix) {
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                System.out.print(matrix[i][j] + " ");
            }
            System.out.println();
        }
    }
}

**Print result:**

//Initial matrix: 
0.1 0.1 
0.1 0.1 
//Same matrix after DMatrixRMaj: 
0.1 0.1 
0.1 0.1