06 2021 档案

HGP-SL程序理解
摘要：一、导入自己的数据集 PyTorch 所有的数据集对象都是torch.utils.data.Dataset的子类。在继承它的时候必须要重写其__len__和__getitem__方法； 为了方便数据的存储和读入，可以将数据存为.pt文件（PyTorch 的标准数据文件）； 四个基本函数 torch_ 阅读全文
posted @ 2021-06-10 15:19 sushamu 阅读(117) 评论(0) 推荐(0) 编辑
diffpool程序理解
摘要：1.np.vstack() 将数组沿垂直方向堆叠起来 np.hstack() 将数组沿水平方向堆叠起来 2.numpy.argmax(a, axis=None, out=None) 返回沿轴axis最大值的索引号。axis=0按列输出，axis=1按行输出。 3.torch.cat(A,B,dim= 阅读全文
posted @ 2021-06-08 13:06 sushamu 阅读(343) 评论(0) 推荐(0) 编辑