pytorch
Bagba
Doctor of Philosophy, machine learning on graphs. ~ 与人分享 是 最好的学习方式。
展开
-
pytorch安装老版本
安装老版本pytorch原创 2023-06-11 16:49:51 · 406 阅读 · 0 评论 -
pytorch load_state_dict不生效
有时候我们会发现,pytorch load一个保存好的模型dict到一个新的模型对象的时候,并没有加载dict里面的数据。原因其实是因为.cuda()原创 2022-07-15 13:53:02 · 1111 阅读 · 0 评论 -
torch.where()详解
torch.where()详解原创 2022-07-08 22:56:04 · 1518 阅读 · 2 评论 -
pytorch reshape view性能对比 (以及einsum, matmul)
pytorch reshap和view性能对比,einsum和matmul性能对比原创 2022-07-07 16:05:55 · 1043 阅读 · 0 评论 -
pytorch dense to coo, edge_index to coo
pytorch 的dense matrix如何转成coo的sparse tensor?给定边的下标,如何转成coo格式的稀疏矩阵?稀疏矩阵如何转成dense的矩阵?原创 2022-07-03 15:08:29 · 767 阅读 · 0 评论 -
jupyter notebook不重启重新import
如何不重启jupyter的情况下,能够重新加载更改过的import的包?很简单,只需要用importlib即可。原创 2022-02-16 15:13:32 · 1960 阅读 · 0 评论 -
pytorch 邻接矩阵转稀疏矩阵 (dense matrix to coo matrix)
如何将一个dense矩阵,比如:一个 N×NN \times NN×N 的邻接矩阵a 转成 pytorch的sparse coo 矩阵。方法一:import numpy as npimport torcha = np.array([[0, 1.2, 0],[2, 3.1, 0],[0.5, 0, 0]])idx = a.nonzero() # (row, col)data = a[idx]# to torch tensoridx_t = torch.LongTensor(np.vstac原创 2022-02-03 15:01:13 · 4509 阅读 · 0 评论 -
jupyter notebook使用argparse(传入list参数即可)
Jupyter notebook其实是可以使用argparse来传入参数的,只需要传入一个list类型的参数即可。原创 2022-01-04 15:36:57 · 2158 阅读 · 0 评论 -
jupyter notebook使用argparse(传入list参数即可)
jupyter notebook 使用argparse,很简单,传入list类型的参数到parser.parse_args()即可。原创 2022-01-05 13:23:11 · 2391 阅读 · 0 评论 -
pytorch einsum, numpy einsum
什么是einsum?爱因斯坦求和约定:https://zhuanlan.zhihu.com/p/101157166https://en.wikipedia.org/wiki/Einstein_notation为什么用?简洁, 强大具体例子pytorch和numpy一样的,这里以pytorch为例.矩阵乘矩阵, C = A x Beg1. (ij, jk -> ik), 实际上就是[C]ik=∑j[A]ij×[B]jk[C]_{ik} = \sum_{j} [A]_{ij} \t原创 2021-04-19 15:16:35 · 509 阅读 · 1 评论 -
查看GPU, cuda,显卡被哪个进程占用
linux查看GPU被哪个进程占用:一个命令搞定。原创 2021-01-25 17:20:24 · 41023 阅读 · 6 评论 -
Pytorch Tensor交换维度
使用transpose,只能交换某两个维度大于2维的tensor的话,也可以想象成二维矩阵的转置操作,可以观察下面的例子得出,之后会画个图说明。>>> t = torch.arange(8).reshape(2,2,2)>>> ttensor([[[0, 1], [2, 3]], [[4, 5], [6, 7]]])>> t.transpose(0,2)tensor([[[0, 4],原创 2020-07-02 10:01:57 · 5801 阅读 · 0 评论 -
Pytorch List Tensor转Tensor,reshape拼接等操作
Pytorch List Tensor转Tensor,reshape拼接等操作持续更新一些常用的Tensor操作,比如List,Numpy,Tensor之间的转换,Tensor的拼接,维度的变换等操作。其它Tensor操作如 einsum等见:待更新。List Tensor转Tensor原创 2020-06-30 17:47:53 · 20788 阅读 · 0 评论