- 博客(2)
- 收藏
- 关注
RSS订阅原创 pytorch的gather函数的一些粗略的理解
pytorch的gather函数的一些理解先给出官方文档的解释，我觉得官方的文档写的已经很清楚了，四个参数分别是input，dim，index，out，输出的tensor是以index为大小的tensor。其中，这就是最关键的定义out[i][j][k] = tensor[index[i][j][k]][j][k] # dim=0out[i][j][k] = tensor[i][ind...
2020-04-28 11:20:26
254
原创 莫烦pytorch教程的CNN代码的一些笔记
莫烦pytorch教程的CNN代码的一些旧版的修改作为一个代码小白，最近在学习莫烦的pytorch教程，因为时间比较长了，有些地方需要做一些修改，写个日记记录下我的笔记。导入包和数据预处理import torchimport torch.nn as nnimport torch.utils.data as Dataimport torchvisionimport matplotlib...
2020-04-25 17:50:52
657
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人