qq_44823873的博客

本博客用于记录pytorch中一些自己比较难理解的点，参考了一下优秀的博客~ 1.关于转置卷积 https://blog.youkuaiyun.com/bu_fo/article/details/109740155?utm_medium=distribute.pc_relevant.none-task-blog-title-3&spm=1001.2101.3001.4242 ...

自己复习: torch.max()使用讲解 torch.max(input, dim) 函数 输入 1.input是softmax函数输出的一个tensor 2.dim是max函数索引的维度0/1，0是每列的最大值，1是每行的最大值 输出 函数会返回两个tensor : 第一个tensor是每行的最大值value 第二个tensor是每行最大值的索引index eg : import torch x = torch.rand(3,4) print('x:\n',x) x: tensor([[0.9300,