pytorch
西工大苗苗
这个作者很懒,什么都没留下…
展开
-
torch.nn.Conv1d及一维卷积详解
近日在搞wavenet,期间遇到了一维卷积,在这里对一维卷积以及其pytorch中的API进行总结,方便下次使用之前对二维卷积是比较熟悉的,在初次接触一维卷积的时候,我以为是一个一维的卷积核在一条线上做卷积,但是这种理解是错的,一维卷积不代表卷积核只有一维,也不代表被卷积的feature也是一维。一维的意思是说卷积的方向是一维的。下边首先看一个简单的一维卷积的例子(batchsize是1,也只...原创 2019-11-06 16:25:10 · 44593 阅读 · 24 评论 -
Pytorch-nn.CrossEntropyLoss
loss_func = nn.CrossEntropyLoss()这个交叉熵损失函数输入的两个参数的shape并不是相同的,一个是各个类别分别的概率(类似于独热码),另一个是位置下标的阿拉伯数字另外:例如: >>> loss = nn.CrossEntropyLoss() >>> input = torch.randn(3,...原创 2019-02-28 11:14:07 · 4273 阅读 · 0 评论 -
pytorch-squeeze中遇到的问题:squeeze失败
一个莫名其妙的错误: data = output[0] data = torch.squeeze(data) print(data.shape)Output:torch.Size([28, 28]) data = output[0] output[0] = torch.sq...原创 2019-02-28 11:18:03 · 1521 阅读 · 0 评论 -
Pytorch遇到的一些问题以及解决方案
Traceback (most recent call last): File "DCGAN.py", line 241, in <module> for x,y in enumerate(dataloader,0): File "/home/rancho/anaconda3/lib/python3.6/site-packages/torch/utils/data/da...原创 2019-03-05 15:21:44 · 3998 阅读 · 0 评论 -
Pytorch-Tensor&Variable
w1 = torch.Tensor([2]) #认为w1 与 w2 是函数f1 与 f2的参数>>> type(w1)<class 'torch.Tensor'>w1 = Variable(w1,requires_grad=True)>>> type(w1)<class 'torch.Tensor', requires_grad=Tr...原创 2019-03-03 20:02:31 · 95 阅读 · 0 评论