pytoch学习
Tcotyledons
这个作者很懒,什么都没留下…
展开
-
pytorch中nn.softmax的dim=1和dim=2的操作
在学习注意力机制的时候,可能需要用到nn.softmax来进行矩阵的乘法结果进行权重归一,但是对于nn.softmax的参数dim不知道具体的操作流程,于是自分别创建两个函数,对应dim=1和dim=2的时候的情况。然后创建一个实例tmp,它的shape是[batch, n, m],所有的操作对不同的batch都是独立进行的。然后开始扔进函数里面,并打印,得到如下。tmp是长这个样子的。原创 2022-09-16 17:04:21 · 1927 阅读 · 0 评论 -
pytorch中DataLoader读不出数据但是又没有报错的情况
今天学习pytorch,数据预处理的时候,将dataset的数据打包之,用dataloader进行封装后,进行测试,但是却一直读不出数据的数值,在此记录一下,虽然感觉是比较小的问题,但是大家在代码调试的时候可能也会先让程序读取小部分的数据看看输出结果,希望大家可以避开这个小问题。原创 2022-09-16 12:57:48 · 1801 阅读 · 0 评论 -
Pytorch中Dataset的一个简单demo
pytorch的Dataset的简单使用原创 2022-09-07 16:07:29 · 396 阅读 · 0 评论