- 博客(3)
- 收藏
- 关注
RSS订阅原创 关于torch.argmax()的在深度学习中应用的理解
今天在debug过程中看到代码中使用了torch.argmax()这个函数。之前在用的时候只是知道返回最大值对应的位置索引，但没有深入思考其含义。在查了大量资料之后，发现很多博客仅仅是照搬了官方文档中的释义，并没有对该函数的使用场景及含义做过多的解释，导致很多初学者在查阅这一函数的用法时一头雾水。下面就结合自己的一些理解对这个函数进行一个直观的解释。在深度学习中，torch.argmax()这个函数最常见的用处就是在图像分类任务下，筛选出网络预测值Prediction中最大概率值对应的class I
2022-01-10 01:19:44
2807
3
原创 torch.nn.ReflectionPad2d()的用法简介
最近在看Pix2PixHD的源码时，发现里面有几处用了nn.ReflectionPad2d()这个函数，查阅了官方文档和网上的资料后，终于弄明白了这个函数的用处。函数用途：对输入图像以最外围像素为对称轴，做四周的轴对称镜像填充。填充顺序：左->右->上->下对于一个4维的Tensor，当只指定一个padding参数时，则表示对四周采用相同的填充行数。# 对四周都填充3行nn.ReflectionPad2d(3) 以下为计算实例：>>...
2021-09-08 16:14:25
12457
8
原创 VSCode设置注释中文字体不斜体并设置颜色
以One Dark Pro主题为例：首先在主题首页可以看到该主题的标识符为zhuangtongfa.material-theme; 在路径 C:\Users\$UserName$\.vscode\extensions\目录下可以看到一个名为"zhuangtongfa.material-theme-版本号"的文件夹，我的版本为3.11.4。 PS:若是在目录下发现两个文件夹，则选择版本最新的那个文件夹。 打开该目录下的themes/OneDark-Pro.json文件，定..
2021-09-01 16:43:49
4225
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人