![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch学习
_sober_
这个作者很懒,什么都没留下…
展开
-
Task07 图预测任务实践
1 知识梳理1.1 Dataset基类继承时需实现的方法:raw_file_names()、processed_file_names()、download()、process()额外需要实现的方法:len()获取数据集中样本数量,get()实现加载单个图的操作1.2 图样本封装成batch和DataLoader类基本思路:将小图的邻接矩阵存储在大图邻接矩阵的对角线上优势:不需要修改GNN算法没有额外计算或内存开销1.3 小图的属性增值与拼接节点序号增值:修改__inc__(原创 2021-07-11 23:46:02 · 135 阅读 · 0 评论 -
jupyter notebook创建虚拟环境与删除虚拟环境
1、创建虚拟环境:condacreate-n 环境名 python=版本号例如创建一个叫做dl的虚拟环境,python版本为3.7:condacreate-n dl python=3.72、激活虚拟环境:conda activate 环境名3、在虚拟环境中安装ipykernel (切记这一步是在对应的虚拟环境中进行):pip install ipykernel使用ipykernel生成虚拟环境的kernel(这一步也是在虚拟环境中进行)python -m ipykernel install原创 2020-11-03 21:00:32 · 3202 阅读 · 0 评论 -
pytorch的gather函数个人理解
首先写一下官方的案例b = torch.Tensor([[1,2,3],[4,5,6]])print bindex_1 = torch.LongTensor([[0,1],[2,0]])index_2 = torch.LongTensor([[0,1,1],[0,0,0]])print torch.gather(b, dim=1, index=index_1)print torch.gather(b, dim=0, index=index_2)这是输出结果: 1 2 3 4 5原创 2020-10-30 23:08:55 · 191 阅读 · 0 评论