![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch相关
娃娃
这个作者很懒,什么都没留下…
展开
-
关于pytorch学习
本博客用于记录pytorch中一些自己比较难理解的点,参考了一下优秀的博客~1.关于转置卷积https://blog.csdn.net/bu_fo/article/details/109740155?utm_medium=distribute.pc_relevant.none-task-blog-title-3&spm=1001.2101.3001.4242...转载 2020-11-27 12:46:32 · 101 阅读 · 0 评论 -
torch.max()
自己复习: torch.max()使用讲解torch.max(input, dim) 函数输入1.input是softmax函数输出的一个tensor2.dim是max函数索引的维度0/1,0是每列的最大值,1是每行的最大值输出函数会返回两个tensor :第一个tensor是每行的最大值value第二个tensor是每行最大值的索引indexeg :import torchx = torch.rand(3,4)print('x:\n',x)x: tensor([[0.9300,原创 2020-11-03 21:08:19 · 658 阅读 · 0 评论