MTL 带状矩阵乘法

原创 2005年03月03日 15:18:00

//MTL 带状矩阵乘法

//整理 by RobinKin
//

#include <iostream>
#include "mtl/matrix.h"
#include "mtl/mtl.h"
#include "mtl/utils.h" 
#include "mtl/linalg_vec.h"

/*

  Sample Output

  Array A in packed form
  [
  [1,1,1,],
  [2,2,2,],
  [3,3,3,],
  [4,4,],
  [5,],
  ]
  x
  [1,2,3,4,5,]
  y
  [1,1,1,1,1,]
  Ax + y
  [15,39,75,75,53,]

  */

using namespace mtl;

typedef matrix< double, banded<>, banded<>, row_major>::type Matrix;
typedef dense1D<double> Vector;

int
main()
{
  const int M = 5;
  const int N = 5;
 
  Matrix A(M, N, 0, 2);

  Vector x(N);
  Vector y(M);

  //           1  1  1              1        1
  //
  //              2  2  2           2        1
  //
  //       A =       3  3  3    x = 3    y = 1
  //
  //                    4  4        4        1
  //
  //                       5        5        1

  int row = 1;
//赋值
  Matrix::iterator ri = A.begin();
  while (ri != A.end()) {
    Matrix::Row::iterator i = (*ri).begin();
    while (i != (*ri).end()) {
      *i = row;
      ++i;
    }
    ++row;
    ++ri;
  }

  for (int i = 0; i < N; ++i)
    x[i] = i + 1;

  mtl::set_value(y, 1);


  std::cout << "Array A in packed form" << std::endl;
  print_row(A);

  std::cout << "x" << std::endl;
  print_vector(x);
  std::cout << "y" << std::endl;
  print_vector(y);

//y=2*x*A+3*y
  mult(A, scaled(x, 2), scaled(y, 3), y);

  std::cout << "Ax + y" << std::endl;
  print_vector(y);

  return 0;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

基于c语言实现的矩阵乘法

  • 2016-03-30 10:21
  • 2.43MB
  • 下载

矩阵乘法.c

poj 2778 DNA Sequence //AC自动机+矩阵乘法

DNA SequenceTime Limit: 1000MS Memory Limit: 65536KTotal Submissions: 5080 Accepted: 1766Description...

n维矩阵乘法

  • 2012-12-11 19:02
  • 220KB
  • 下载

基于C语言的矩阵乘法

Hadoop实例:单轮MapReduce的矩阵乘法Demo

import java.io.IOException; import org.apache.hadoop.conf.Configuration; import org.apache.hadoop....

【BZOJ1898】[Zjoi2005]Swamp 沼泽鳄鱼【DP】【矩阵乘法】

http://www.lydsy.com/JudgeOnline/problem.php?id=1898 论文题,详见《矩阵乘法在信息学中的应用》俞华程。 样例果然弱...提供一组数据: Inpu...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)