稀疏矩阵csc_matrix函数的使用理解

对csc_matrix稀疏矩阵的理解

背景

项目中使用到OSQP求解器,其使用了稀疏矩阵的方式对数据进行存储,使用过程中经常会忘记稀疏矩阵的几个存储数组存储内容的含义,记录一波,此处以图展示的方式来方便理解加深记忆。

以图的方式表示

这里以列存储的方式来说明,列存储方式理解了,行存储方式自然也理解了。下面主要是对三个存储数组的存储方式的理解:

indptr = [ ] , 以累加的方式存储每一列包含有非零数字的个数从0开始
indices = [ ],存储非零数字所在的行索引值
data = [ ], 以列顺序存储所有非零的数字

示意图

在这里插入图片描述

简单例子

在这里插入图片描述

  • 10
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
在PyTorch中,CSC(Compressed Sparse Column)和CSR(Compressed Sparse Row)是用于稀疏矩阵存储和操作的格式。CSC和CSR都是将稀疏矩阵压缩为三个数组,分别存储非零元素的值、列索引和行指针。 要使用CSC和CSR格式,首先需要安装SciPy库,因为PyTorch不直接支持这两种格式。使用以下命令安装SciPy: ```python pip install scipy ``` 安装完成后,可以使用SciPy的`csr_matrix`和`csc_matrix`函数稀疏矩阵转换为CSR和CSC格式。例如,将一个二维稀疏矩阵转换为CSR格式: ```python import torch from scipy.sparse import csr_matrix sparse_matrix = torch.sparse_coo_tensor(indices, values, size) csr_matrix = csr_matrix(sparse_matrix.to_dense().numpy()) ``` 在上面的代码中,`indices`是一个包含非零元素的索引的张量,`values`是一个包含非零元素值的张量,`size`是稀疏矩阵的形状。首先,使用PyTorch的`sparse_coo_tensor`函数创建一个COO(Coordinate)格式的稀疏张量,然后使用`to_dense`方法将其转换为密集张量。最后,使用SciPy的`csr_matrix`函数将密集矩阵转换为CSR格式。 类似地,可以使用`csc_matrix`函数稀疏矩阵转换为CSC格式: ```python from scipy.sparse import csc_matrix csc_matrix = csc_matrix(sparse_matrix.to_dense().numpy()) ``` 转换为CSR或CSC格式后,可以使用SciPy提供的各种函数稀疏矩阵进行操作和计算。 请注意,这里的示例代码是将PyTorch的稀疏张量转换为CSR或CSC格式。如果你已经有一个稀疏矩阵的CSR或CSC表示,可以直接使用相关的SciPy函数进行操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值