如何将具有分数的两个矩阵乘以c中的输入

时间:2016-09-19 16:45:00

标签: c arrays matrix matrix-multiplication

我需要一个c或java代码来将两个输入相乘的矩阵相乘,例如我们有两个2x2矩阵A和B

Matrix A

    2.5 3.5     
    0.5 1 

Matrix B

    2.7 3.8
    0.9 1.5

我需要使用naive和strassen的算法来乘以这些矩阵,所以如果有人能为我提供解决方案,那将是一个很大的帮助。 先感谢您。

2 个答案:

答案 0 :(得分:0)

尝试:

假设矩阵采用以下形式:

Double[][] A={{4.00,3.00},{2.00,1.00}}; 
Double[][] B={{0.50,1.500},{1.0,2.00}};

然后,您可以将其作为参数传递给此函数:

public static Double[][] multiplicar(Double A[][],Double B[][]){
    Double[][] C= new Double[2][2];
    int i,j,k;
     for (i = 0; i < 2; i++) {
         for (j = 0; j < 2; j++) {
             C[i][j] = 0.00000;
         }
     }
    for(i=0;i<2;i++){
        for(j=0;j<2;j++){
            for (k=0;k<2;k++){
                  C[i][j]+=(A[i][k]*B[k][j]);

            }

        }
    }
    return C;
}

答案 1 :(得分:0)

我想用自己的Strassen优化实现矩阵乘法。它是高度优化的,因此几乎没有教学用途,但后来我记得维基百科在Strassen矩阵乘法项中有实际的C代码。

那里的实施并不是最好的:

  • 如果大小低于某个限制,它就没有回归到天真算法
  • 只允许矩阵大小为2的幂
  • 如果与有限浮点数据类型一起使用,它会累积很多错误
  • 它使用了大量内存,超过了必要的
  • 需要一些工作才能使其可并行化

但它有效,所以......

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define MM_OK 1
#define MM_ERR 0

void mat_print(double **M, int dim);
double **mat_init(int dim);
void mat_clear(double **M, int dim);
double **mat_rand(int dim, int seed);
int mat_mul_naive(double **A, double **B, double **C, int dim);
int mat_add(double **A, double **B, double **C, int dim);
int mat_sub(double **A, double **B, double **C, int dim);
int mat_mul_strassen(double **A, double **B, double **C, int dim);

// ALL CHECKS OMMITTED!

// Octave/Matlab format, for getting a second opinion
void mat_print(double **M, int dim)
{
  int i, j;
  printf("[");
  for (i = 0; i < dim; i++) {
    for (j = 0; j < dim; j++) {
      printf("%.10g", M[i][j]);
      if (j < dim - 1) {
        putc(',', stdout);
      }
    }
    if (i < dim - 1) {
      putc(';', stdout);
    }
  }
  printf("]\n");
}

double **mat_init(int dim)
{
  int i, j;
  double **M;
  M = malloc(dim * sizeof(double *));
  if (M == NULL) {
    return NULL;
  }
  for (i = 0; i < dim; i++) {
    M[i] = malloc(dim * sizeof(double));
    if (M[i] == NULL) {
      mat_clear(M, i);
      return NULL;
    }
    for (j = 0; j < dim; j++) {
      M[i][j] = 0;
    }
  }
  return M;
}

void mat_clear(double **M, int dim)
{
  int i;
  for (i = 0; i < dim; i++) {
    free(M[i]);
  }
  free(M);
}

double **mat_rand(int dim, int seed)
{
  int i, j;
  double **M;
  M = mat_init(dim);
  if (M == NULL) {
    return NULL;
  }
  srand(seed);
  for (i = 0; i < dim; i++) {
    for (j = 0; j < dim; j++) {
      M[i][j] = (double) rand() / (double) rand();
    }
  }
  return M;
}

int mat_mul_naive(double **A, double **B, double **C, int dim)
{
  int i, j, k;
  if (C == NULL) {
    return MM_ERR;
  }
  for (i = 0; i < dim; i++) {
    for (k = 0; k < dim; k++) {   //<- exchanged for better
      for (j = 0; j < dim; j++) { //<- memory access
        C[i][j] += (A[i][k] * B[k][j]);
      }
    }
  }
  return MM_OK;
}


// helper for mat_mul_strassen, only square matrices of same size
int mat_add(double **A, double **B, double **C, int dim)
{
  int i, j;
  if (C == NULL) {
    fprintf(stderr, "C == NULL in mat_add\n");
    return MM_ERR;
  }
  for (i = 0; i < dim; i++) {
    for (j = 0; j < dim; j++) {
      C[i][j] = A[i][j] + B[i][j];
    }
  }
  return MM_OK;
}

// helper for mat_mul_strassen, only square matrices of same size
int mat_sub(double **A, double **B, double **C, int dim)
{
  int i, j;
  if (C == NULL) {
    fprintf(stderr, "C == NULL in mat_sub\n");
    return MM_ERR;
  }
  for (i = 0; i < dim; i++) {
    for (j = 0; j < dim; j++) {
      C[i][j] = A[i][j] - B[i][j];
    }
  }
  return MM_OK;
}

// This implementation is from Wikipedia
// https://en.wikipedia.org/w/index.php?title=Strassen_algorithm&oldid=498910018#Source_code_of_the_Strassen_algorithm_in_C_language
// And it is a really slow one.
int mat_mul_strassen(double **A, double **B, double **C, int dim)
{
  if (dim == 1) {
    C[0][0] = A[0][0] * B[0][0];
    return MM_OK;
  }
#ifdef MAKE_IT_ACTUALLY_WORK_FAST
  // 256 on my machine, YMMV
  else if (dim <= 256) {
    return mat_mul_naive(A, B, C, dim);
  }
#endif
  else {
    int new_dim = dim / 2;
    double **a11, **a12, **a21, **a22;
    double **b11, **b12, **b21, **b22;
    double **c11, **c12, **c21, **c22;
    double **p1, **p2, **p3, **p4, **p5, **p6, **p7;
    double **aResult;
    double **bResult;
    int i, j;
    a11 = mat_init(new_dim);
    a12 = mat_init(new_dim);
    a21 = mat_init(new_dim);
    a22 = mat_init(new_dim);

    b11 = mat_init(new_dim);
    b12 = mat_init(new_dim);
    b21 = mat_init(new_dim);
    b22 = mat_init(new_dim);

    c11 = mat_init(new_dim);
    c12 = mat_init(new_dim);
    c21 = mat_init(new_dim);
    c22 = mat_init(new_dim);

    p1 = mat_init(new_dim);
    p2 = mat_init(new_dim);
    p3 = mat_init(new_dim);
    p4 = mat_init(new_dim);
    p5 = mat_init(new_dim);
    p6 = mat_init(new_dim);
    p7 = mat_init(new_dim);

    aResult = mat_init(new_dim);
    bResult = mat_init(new_dim);
    // the divide part of divide&conquer
    for (i = 0; i < new_dim; i++) {
      for (j = 0; j < new_dim; j++) {
        a11[i][j] = A[i][j];
        a12[i][j] = A[i][j + new_dim];
        a21[i][j] = A[i + new_dim][j];
        a22[i][j] = A[i + new_dim][j + new_dim];

        b11[i][j] = B[i][j];
        b12[i][j] = B[i][j + new_dim];
        b21[i][j] = B[i + new_dim][j];
        b22[i][j] = B[i + new_dim][j + new_dim];
      }
    }

    // Calculating p1 to p7:

    mat_add(a11, a22, aResult, new_dim);        // a11 + a22
    mat_add(b11, b22, bResult, new_dim);        // b11 + b22
    mat_mul_strassen(aResult, bResult, p1, new_dim);    // p1 = (a11+a22) * (b11+b22)

    mat_add(a21, a22, aResult, new_dim);        // a21 + a22
    mat_mul_strassen(aResult, b11, p2, new_dim);        // p2 = (a21+a22) * (b11)

    mat_sub(b12, b22, bResult, new_dim);        // b12 - b22
    mat_mul_strassen(a11, bResult, p3, new_dim);        // p3 = (a11) * (b12 - b22)

    mat_sub(b21, b11, bResult, new_dim);        // b21 - b11
    mat_mul_strassen(a22, bResult, p4, new_dim);        // p4 = (a22) * (b21 - b11)

    mat_add(a11, a12, aResult, new_dim);        // a11 + a12
    mat_mul_strassen(aResult, b22, p5, new_dim);        // p5 = (a11+a12) * (b22)  

    mat_sub(a21, a11, aResult, new_dim);        // a21 - a11
    mat_add(b11, b12, bResult, new_dim);        // b11 + b12
    mat_mul_strassen(aResult, bResult, p6, new_dim);    // p6 = (a21-a11) * (b11+b12)

    mat_sub(a12, a22, aResult, new_dim);        // a12 - a22
    mat_add(b21, b22, bResult, new_dim);        // b21 + b22
    mat_mul_strassen(aResult, bResult, p7, new_dim);    // p7 = (a12-a22) * (b21+b22)

    // calculating c21, c21, c11 e c22:

    mat_add(p3, p5, c12, new_dim);      // c12 = p3 + p5
    mat_add(p2, p4, c21, new_dim);      // c21 = p2 + p4

    mat_add(p1, p4, aResult, new_dim);  // p1 + p4
    mat_add(aResult, p7, bResult, new_dim);     // p1 + p4 + p7
    mat_sub(bResult, p5, c11, new_dim); // c11 = p1 + p4 - p5 + p7

    mat_add(p1, p3, aResult, new_dim);  // p1 + p3
    mat_add(aResult, p6, bResult, new_dim);     // p1 + p3 + p6
    mat_sub(bResult, p2, c22, new_dim); // c22 = p1 + p3 - p2 + p6

    // Grouping the results obtained in a single matrix:
    for (i = 0; i < new_dim; i++) {
      for (j = 0; j < new_dim; j++) {
        C[i][j] = c11[i][j];
        C[i][j + new_dim] = c12[i][j];
        C[i + new_dim][j] = c21[i][j];
        C[i + new_dim][j + new_dim] = c22[i][j];
      }
    }
    mat_clear(a11, new_dim);
    mat_clear(a12, new_dim);
    mat_clear(a21, new_dim);
    mat_clear(a22, new_dim);

    mat_clear(b11, new_dim);
    mat_clear(b12, new_dim);
    mat_clear(b21, new_dim);
    mat_clear(b22, new_dim);

    mat_clear(c11, new_dim);
    mat_clear(c12, new_dim);
    mat_clear(c21, new_dim);
    mat_clear(c22, new_dim);

    mat_clear(p1, new_dim);
    mat_clear(p2, new_dim);
    mat_clear(p3, new_dim);
    mat_clear(p4, new_dim);
    mat_clear(p5, new_dim);
    mat_clear(p6, new_dim);
    mat_clear(p7, new_dim);
    mat_clear(aResult, new_dim);
    mat_clear(bResult, new_dim);

  }
  return MM_OK;
}

static int ispow2(unsigned int x)
{
  while (((x & 1) == 0) && x > 1) {
    x >>= 1;
  }
  if (x == 1) {
    return 1;
  }
  return 0;
}

#include <time.h>
int main(int argc, char **argv)
{
  //int res;
  int dim1;
  double **A, **B, **C, **D;
  clock_t start, stop;

  // only square matrices of same size
  if (argc != 2) {
    fprintf(stderr, "Usage: %s dimension\n", argv[0]);
    exit(EXIT_FAILURE);
  }

  dim1 = atoi(argv[1]);
  if (!ispow2(dim1)) {
    fprintf(stderr,
            "Dimension of matrix must be a power of two and %d is not\n", dim1);
    exit(EXIT_FAILURE);
  }

  A = mat_rand(dim1, 123);
  B = mat_rand(dim1, 456);

  //mat_print(A, dim1);
  //mat_print(B, dim1);

  C = mat_init(dim1);
  start = clock();
  mat_mul_naive(A, B, C, dim1);
  stop = clock();
  printf("Naive    %.10f seconds\n", (double) (stop - start) / CLOCKS_PER_SEC);
  //mat_print(C, dim1);

  D = mat_init(dim1);
  start = clock();
  mat_mul_strassen(A, B, D, dim1);
  stop = clock();
  printf("Strassen %.10f seconds\n", (double) (stop - start) / CLOCKS_PER_SEC);
  //mat_print(D, dim1);

  mat_sub(C, D, C, dim1);
  // results will differ more and more the larger the matrix is
  //mat_print(C, dim1);

  mat_clear(A, dim1);
  mat_clear(B, dim1);
  mat_clear(C, dim1);
  mat_clear(D, dim1);
  exit(EXIT_SUCCESS);
}