稀疏矩阵转CSR格式存储

1.随机生成稀疏矩阵

#include<stdlib.h>
#include<assert.h>
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;void fill_random(float*data, int m, int n) {srand((unsigned) (time(NULL))); //每次生成的随机数不一样for (int i = 0; i < m*n; i++) {data[i] = rand() % 100;  //生成100以内的随机数if (data[i] < 80) //made sparse data[i] = 0;}
}void print_matrix(float*data, int m, int n) {for (int i = 0; i < m; i++) {for (int j = 0; j < n; j++) {int l = i + j * m;cout << data[l] << " a";}cout << endl;} cout << endl;
}void print_matrix(int*data, int m, int n) {for (int i = 0; i < m; i++) {for (int j = 0; j < n; j++) {int l = i + j * m;cout << data[l] << " b";}cout << endl;}cout << endl;
}void dense2csr(float*data, int*&rowPtr, int*&colInd, float*&val, int m, int n) {rowPtr = (int*)malloc(sizeof(int)*(m + 1));int* tcolInd = (int*)malloc(sizeof(int)*(m *n));float* tval = (float*)malloc(sizeof(float)*(m *n));int towtal = m * n;int nnv = 0;for (int i = 0; i < m; i++) {rowPtr[i] = nnv;for (int j = 0; j < n; j++) {int l = i + j * m;if (data[l] != 0) {tcolInd[nnv] = j;tval[nnv] = data[l];nnv++;}}}rowPtr[m] = nnv;colInd = (int*)malloc(sizeof(int)*(nnv));val = (float*)malloc(sizeof(float)*(nnv));memcpy(colInd, tcolInd, sizeof(float)*nnv);memcpy(val, tval, sizeof(float)*nnv);free(tcolInd);free(tval);
}void csr2dense(int*&rowPtr, int*&colInd, float*&val, float*&data, int m, int n) {data = (float*)calloc(m*n,sizeof(float));int nnv = rowPtr[m];int* help = (int*)malloc(sizeof(int)*nnv); //indicates row index of elementfor (int i = 0, l = 1; i < nnv; i++) {while (i >= rowPtr[l])l++;help[i] = l-1;}for (int i = 0; i < nnv; i++) {int datal = help[i] + colInd[i] * m;data[datal] = val[i];}free(help);
}bool same_matrix(float*A, float*B, int m, int n) {for (int i = 0; i < m*n; i++) {if (A[i] != B[i])return false;}return true;
}int main() {int m = 5;int n = 5;float*A = (float*)malloc(sizeof(float)*m*n);fill_random(A, m, n);print_matrix(A, m, n);int*csrRowPtr;int*csrColInd;float*csrVal;dense2csr(A, csrRowPtr, csrColInd, csrVal, m, n);print_matrix(csrRowPtr, 1, m+1);print_matrix(csrColInd, 1, csrRowPtr[m]);print_matrix(csrVal, 1, csrRowPtr[m]);float*newA;csr2dense(csrRowPtr, csrColInd, csrVal, newA, m, n);print_matrix(newA, m, n);assert(same_matrix(A, newA, m, n)); //debug mode
}

稀疏矩阵转CSR格式存储相关推荐

在python中使用json格式存储数据
在python中使用json格式存储数据代码如下: import jsonlist1 = [{'A': [1, 2, 3, 4, 5, 6], 'B': [3, 4, 5, 6, 7]},{'C': ...
5.3稀疏矩阵的十字链表存储
十字链表产生原因:当矩阵的非零元个数和位置在操作过程中变化较大时,就不宜采用顺序存储结构来表示三元组的线性表. 十字链表特点: 每一个非零元开用含5个域的结点表示,其中i.j和e这3个域分别表示该非零 ...
稀疏矩阵的十字链表存储表示和实现（第五章 P104 算法5.4）
稀疏矩阵的十字链表存储当矩阵的非零元个数和位置在操作过程中变化较大时,就不宜采用顺序存储结构来表示三元组的线性表.对这种类型的矩阵,采用链式存储结构表示三元组的线性表更为恰当. 在链表中,每个非零元 ...
cuda编程与gpu并行计算（六）：图稀疏矩阵转为CSR结构并传入gpu
虽然sepgraph有这部分代码,还是自己先试着实现一下,这样读起来也方便行压缩格式Compressed Sparse Row (CSR) CSR需要三种数据来表达:数值.列号.行转移.CSR不是三 ...
爬取中国大学排名并以csv格式存储
爬取中国大学排名并以csv格式存储 import requests from bs4 import BeautifulSoup import bs4def get_content(url):try:u ...
plt.imshow()显示以ndarray格式存储的图像数据
1 plt.imshow()详细介绍:常用参数为红色标注输入可以是实际的RGB(A)数据,也可以是2D标量数据,这些数据将被渲染为伪彩色图像.为了显示灰度图像,使用参数设置颜色映射:cmap='gr ...
Mysql中使用json格式存储数据好吗？
在最近的一次项目开发过程中,在数据表设计阶段,对是否用json格式存储某些数据我们产生了分歧.以往项目中对此点比较随意,导致数据表中有些json格式数据体积很大,层次很深,我担心这会降低数据查询和解析 ...
下面是以十六进制格式存储的一个 UDP 首部：~~~TCP连接使用1000字节的窗口值，而上一次的确认号是22001~~那么下一个报文段的序号是否就是 x + 1 呢？在本题中列出的 8 种情况下，画
5-10 试说明运输层中伪首部的作用用于计算运输层数据报校验和 5-11 某个应用进程使用运输层的用户数据报UDP,然而继续向下交给IP层后,又封装成IP数据报.既然都是数据报,可否跳过UDP而直接 ...
在MYSQL里使用哪种格式存储图片？之解惑总结
2019独角兽企业重金招聘Python工程师标准>>> 1 写入时并无格式之分,只是保存二进制数据,读取后和写入时的格式一样. jpg图片用二进制格式写入mysql,再以二进制格式读 ...

稀疏矩阵转CSR格式存储

稀疏矩阵转CSR格式存储

稀疏矩阵转CSR格式存储相关推荐

最新文章

热门文章