首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Java矩阵运算,并行colt矩阵-矩阵乘法

Java矩阵运算,并行colt矩阵-矩阵乘法
EN

Stack Overflow用户
提问于 2013-04-02 09:26:35
回答 1查看 1.4K关注 0票数 1

我在java中使用Jama进行矩阵运算,但是因为它们没有稀疏矩阵,所以我开始使用Parallel Cold Library (PColt)。它是Colt的多线程版本。我试着将两个方阵相乘,AxB (它是矩阵-矩阵乘法,而不是元素(或标量)乘法),size (NxN).I在PColt中找不到提供的矩阵-矩阵乘法(我不想要元素乘法),所以我将该方法编写为以下代码。当我为N=1000乘以两个矩阵时,它几乎需要5分钟才能完成。如果有人知道如何在pcolt中将两个矩阵相乘,那将非常感谢。或者,如果您看到代码中有任何不必要的错误,并使复杂性很高,请通知我。我的方法如下:

代码语言:javascript
复制
    /**
 * Linear algebraic matrix-matrix multiplication; (new)A = A x B. A[i,j] =
 * Sum(A[i,k] * B[k,j]), k=0..n-1. Matrix shapes: A(m x n), B(n x p), A(m x
 * p).
 * 
 * @param matrix1
 *            first matrix
 * @param matrix2
 *            second matrix
 * @return changes matrix1 with new values matrix-matrix multiplication
 */

public static FloatMatrix2D multiplyMatrix(FloatMatrix2D matrix1,
        FloatMatrix2D matrix2) {
    // create a matrix same size of input matrix
    FloatMatrix2D resultMatrix = matrix1.like();
    // matrix-matrix multiplication row of first matrix must be equal to
    // column of second matrix
    if (matrix1.rows() == matrix2.columns()) {
        for (int i = 0; i < matrix1.rows(); i++) {
            for (int j = 0; j < matrix2.columns(); j++) {
                FloatMatrix1D rowVec = getRow(matrix1, i).copy();
                FloatMatrix1D colVec = getCol(matrix2, j).copy();
                // first multiplies each row with each column and then sum
                // up the result and assign the result to value of matrix
                float multOfVects = rowVec.assign(colVec,
                        FloatFunctions.mult).aggregate(FloatFunctions.plus,
                        FloatFunctions.identity);
                // set sum of vectors to the new matrix
                resultMatrix.setQuick(i, j, multOfVects);
                
            }
            System.out.println(" i th row "+ i);
        }
    } else {
        System.err
                .println("Row size of first matrix must be equal to Column size of second matrix: "
                        + matrix1 + " != " + matrix2);
    }

    return resultMatrix;
}

解决方案..。

好的多奇我有办法了。实际上忘记上面的代码。PColt提供了矩阵-矩阵乘法,但方法名称令人困惑。要将两个矩阵相乘,请使用以下方法:

代码语言:javascript
复制
public DoubleMatrix2D zMult(DoubleMatrix2D B,
                            DoubleMatrix2D C)

线性代数矩阵-矩阵乘法;

这里,要注意参数的顺序,因为结果会保存到最后一个参数C。也可以将C设置为null,这会返回一个新的矩阵。

EN

回答 1

Stack Overflow用户

发布于 2014-01-22 18:46:19

矩阵乘法方法在DoubleAlgebra类中。

代码语言:javascript
复制
DenseDoubleAlgebra algebra = new DenseDoubleAlgebra();
DoubleMatrix2D productMatrix = algebra.mult(aMatrix, anotherMatrix);
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/15754643

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档