PyG|邻接矩阵转为coo

在使用PyG框架的时候，PyG要求输入的是 edge_index 格式，而不是我们所使用的邻接矩阵格式,即N x N。

import scipy.sparse as sp
import numpy as np
import torch# adj_matrix 是邻接矩阵
tmp_coo = sp.coo_matrix(adj_matrix)
values = tmp_coo.data
indices = np.vstack((tmp_coo.row,tmp_coo.col))
i = torch.LongTensor(indices)
v = torch.LongTensor(values)
edge_index=torch.sparse_coo_tensor(i,v,tmp_coo.shape)

PyG|邻接矩阵转为coo_matrix格式相关推荐

读取mysql数据库的数据，转为json格式
# coding=utf-8 ''' Created on 2016-10-26 @author: Jennifer Project:读取mysql数据库的数据,转为json格式 ''' import ...
Convert PLY to VTK Using PCL 1.6.0 or PCL 1.8.0 使用PCL库将PLY格式转为VTK格式
PLY格式是比较流行的保存点云Point Cloud的格式,可以用MeshLab等软件打开,而VTK是医学图像处理中比较常用的格式,可以使用VTK库和ITK库进行更加复杂的运算处理.我们可以使用Par ...
Sql：成功解决将sql输出的datetime时间格式转为常规格式
Sql:成功解决将sql输出的datetime时间格式转为常规格式目录 SQL应用案例 1.将sql输出的datetime时间格式转为常规格式 SQL应用案例 1.将sql输出的datetime时间 ...
通过输入流获取json格式数据转为map格式
将通过输入流获取json格式数据转为map格式转换代码: try {int length = 0;//通过输入流获取到json格式数据ServletInputStream inputStream = ...
Python实例 -- .git格式转为.png格式
前言最近在跑JRST肺部数据集相关实验的时候,发现JRST数据集的masks的格式为.gif格式的,然而我之前所有的代码都是对.png格式的图片进行操作的,故需要将.gif格式的图片转为.png格式 ...
【QGIS入门实战精品教程】4.7：QGIS如何将矢量数据转为GeoJSON格式？
本文以案例的形式,讲述在QGIS专业软件中,将矢量数据转为GeoJSON的方法. 扩展阅读: [ArcGIS风暴]如何将矢量数据(点.线.面)折点坐标转为GeoJSON格式? 在QGIS中,可以直接将 ...
itextPdf~将PDF页面大小转为A4格式
本文章向大家介绍将PDF页面大小转为A4格式---itextPdf实现,主要包括将PDF页面大小转为A4格式---itextPdf实现使用实例.应用技巧.基本知识点总结和需要注意事项,具有一定的参考价 ...
python输出jpg_利用python实现.dcm格式图像转为.jpg格式
如下所示: import pydicom import matplotlib.pyplot as plt import scipy.misc import pandas as pd import nu ...
python生成json_如何将Python数组转为Json格式数据并存储？
在Python中将数组转为Json数据存储时需要用到将json模块中的json.dumps()或者json.dump()方法. ·json.dumps()方法用法使用json.dumps( )方法将 ...

PyG|邻接矩阵转为coo_matrix格式

PyG|邻接矩阵转为coo_matrix格式相关推荐

最新文章

热门文章