WebMasterZ的博客

pytorch的gather函数的一些理解 先给出官方文档的解释，我觉得官方的文档写的已经很清楚了，四个参数分别是input，dim，index，out，输出的tensor是以index为大小的tensor。 其中，这就是最关键的定义 out[i][j][k] = tensor[index[i][j][k]][j][k] # dim=0 out[i][j][k] = tensor[i][ind...

莫烦pytorch教程的CNN代码的一些旧版的修改 作为一个代码小白，最近在学习莫烦的pytorch教程，因为时间比较长了，有些地方需要做一些修改，写个日记记录下我的笔记。 导入包和数据预处理 import torch import torch.nn as nn import torch.utils.data as Data import torchvision import matplotlib...