机器学习
swy_swy_swy
这个作者很懒,什么都没留下…
展开
-
view size is not compatible with input tensor‘s size and stride
原因:对tensor使用view()时要求tensor在内存中地址是连续的,但有时不满足这个条件。解决方法:将地址变为连续的。在.view()前加上.contiguous()。在对pytorch tensor使用view()方法时报错。原创 2022-11-21 12:35:10 · 536 阅读 · 0 评论 -
conda安装pytorch_geometric
欲安装pytorch_geometric包,网上看几个帖子都是先安装几个(一般是四个)包云云,但查看官网。原创 2022-11-17 00:41:09 · 558 阅读 · 0 评论 -
numpy array 升维
numpy array 升维训练模型的时候,有时给你的数据并不完全与模型期待的输入匹配。比如最近做一个医疗影像相关的项目,给我的数据是三维的MRI, 模型的输入却是四维, 多了一维channel。此时,我需要在数据的首个维度之前加上一个channel维度。代码示例我们以二维数组升到三维为例import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6]])print(arr.shape) # (2, 3)arr_1 = ar原创 2021-11-11 13:19:44 · 1979 阅读 · 0 评论