在Java中查找复杂矩阵的SVD矩阵

问题描述 投票:1回答:1

我目前正致力于音频信号处理项目,需要在Java中的复杂矩阵上使用SVD。我目前的线性代数库是Apache Commons。然而,它只提供真实矩阵的SVD,JAMA,JBLAS,EJML,ojAlgo也不支持复杂的SVD。

我一直在使用一些技巧,使用找到的here技术从等效的真实矩阵中找到SVD。然而,当我重建矩阵时,该技术对于虚部而言具有相当大的不准确性。

如果有人使用真正的SVD库或在java中支持复杂SVD的库找到复杂的SVD,我将非常感谢。


这是我一直在做的事情:

    //Array2DRowFieldMatrix<Complex> A = some matrix defined earlier
    Array2DRowRealMatrix AA = new Array2DRowRealMatrix(2 * row, 2 * col);
    Complex Aentry;

    for (int c = 0; c < row; c++) {
        for (int s = 0; s < col; s++) {
            Aentry = A.getEntry(c, s);

            AA.setEntry(c, s, Aentry.getReal());
            AA.setEntry(c, col + s, -Aentry.getImaginary());
            AA.setEntry(row + c, s, Aentry.getImaginary());
            AA.setEntry(row + c, col + s, Aentry.getReal());
        }
    }

    Array2DRowRealMatrix UU, SS, VV;

    svd = new SingularValueDecomposition(AA);

    UU = (Array2DRowRealMatrix) svd.getU();
    SS = (Array2DRowRealMatrix) svd.getS();
    VV = (Array2DRowRealMatrix) svd.getV();

    double[][] tempU = new double[row][2 * row];
    double[][] tempV = new double[col][2 * row];
    double[] tempS = new double[row];

    Array2DRowFieldMatrix<Complex> U = new Array2DRowFieldMatrix<>(ComplexField.getInstance(), row, row);
    Array2DRowFieldMatrix<Complex> S = new Array2DRowFieldMatrix<>(ComplexField.getInstance(), row, row);
    Array2DRowFieldMatrix<Complex> V = new Array2DRowFieldMatrix<>(ComplexField.getInstance(), col, row);
    Array2DRowFieldMatrix<Complex> recon, diff;

    UU.copySubMatrix(row, 2 * row - 1, 0, 2 * row - 1, tempU);
    VV.copySubMatrix(col, 2 * col - 1, 0, 2 * row - 1, tempV);

    for (int i = 0; i < row; i++) {
        for (int j = 0; j < row; j++) {
            U.setEntry(i, j, new Complex(tempU[i][2 * j], tempU[i][2 * j + 1]));
        }
    }

    for (int i = 0; i < col; i++) {
        for (int j = 0; j < row; j++) {
            V.setEntry(i, j, new Complex(tempV[i][2 * j], tempV[i][2 * j + 1]));
        }
    }

    for (int i = 0; i < row; i++) {
        tempS[i] = SS.getEntry(i * row, i * row);
        if (tempS[i] == 0) {
            tempS[i] = EPSILON;
        }
    }

    for (int i = 0; i < row; i++) {
        for (int j = 0; j < row; j++) {
            if (i != j) {
                S.setEntry(i, j, Complex.ZERO);
            }
        }

        S.setEntry(i, i, new Complex(tempS[i]));
    }

    recon = (Array2DRowFieldMatrix<Complex>)U.multiply(S).multiply(conjugate(V).transpose());
java android matrix linear-algebra svd
1个回答
0
投票

如果您可以使用本机库,那么jeigen是一个选项

© www.soinside.com 2019 - 2024. All rights reserved.