一. dense矩阵转coo稀疏矩阵

    def dense_to_coo(a:torch.Tensor):idx = torch.nonzero(a).Tdata = a[idx[0],idx[1]]coo = torch.sparse_coo_tensor(idx, data, a.shape)return coo

二. edge_index转coo稀疏矩阵

这里的edge index也就是边的下标，indices。

    def edge_index_to_coo(edge_index:torch.Tensor, N:int):v = torch.ones(edge_index.size(1))coo = torch.sparse_coo_tensor(edge_index, v, (N, N))return coo

三. coo转dense

dense = coo.to_dense()即可。

pytorch dense to coo, edge_index to coo相关推荐

python中稀疏矩阵的常用表示COO LIL CSR CSC【上篇】
前言 sklearn调用独热编码函数encoder.fit_transform()返回的是scipy.sparse._csr.csr_matrix类型. torch的tensor也是一种matrix, ...
CEO, CFO,CTO,COO的含义？
CEO:Chief Executive Officer (首席执行官) CFO :Chief financial Officer (首席财政官或财务总监) CTO:Chief technology o ...
CEO、COO、CFO、CTO、CIO、CBO、CDO……日常必知
CEO(Chief executive officer)首席执行官 CTO(Chief technology officer)首席技术官 CIO(Chief information officer)首 ...
GNN PyTorch functions
目录 PyTorch fundamental functions torch.nonzero()方法 torch.no_grad()函数 torch.state_dict()函数 torch.wher ...
pytorch笔记：torch.sparse类
PyTorch 提供了 torch.Tensor 来表示一个包含单一数据类型元素的多维数组. 默认情况下,数组元素连续存储在内存中,从而可以有效地实现各种数组处理算法,这些算法依赖于对数组元素的快速访 ...
Pytorch torch.sparse_coo_tensor()
Pytorch 构造稀疏 Tensor torch.sparse_coo_tensor(indices, values,size=None, *,dtype=None,device=None,requ ...
PyG文档之二：快速入门
快速入门图数据常规操作栗子1:实例化一个data 栗子2:Data的一些内置函数标准数据集栗子3:载入标准数据集小批量训练栗子4:PyG的DataLoader 数据转换栗子5:PyG 的 ...
稀疏矩阵三元组严蔚敏_Sparse稀疏矩阵主要存储格式总结
在数据科学和深度学习等领域会采用矩阵来存储数据,但当矩阵较为庞大且非零元素较少时,运算效率和存储有效率并不高.所以,一般情况我们采用Sparse稀疏矩阵的方式来存储矩阵,来提高存储和运算效率.下面将对 ...
scipy.sparse、pandas.sparse、sklearn稀疏矩阵的使用
单机环境下,如果特征较为稀疏且矩阵较大,那么就会出现内存问题,如果不上分布式 + 不用Mars/Dask/CuPy等工具,那么稀疏矩阵就是一条比较容易实现的路. 文章目录 1 scipy.sparse ...

pytorch dense to coo, edge_index to coo

一. dense矩阵转coo稀疏矩阵

二. edge_index转coo稀疏矩阵

三. coo转dense

pytorch dense to coo, edge_index to coo相关推荐

最新文章

热门文章