torch.nn
文章平均质量分 95
disanda
disanda.github.io
展开
-
Pytorch中的卷积与反卷积详解(conv2d和convTranspose2d)
Pytorch中的卷积与反卷积详解(conv2d和convTranspose2d)卷积和反卷积是图片计算在深度学习中常用的上采样和下采样操作。相比其他采样操作,卷积计算不仅可以保存参数的梯度传递(适用用BP),还可以改变图片的通道以更好的整合局部特征。在torchn.nn中,卷积操作是一个函数,输入为一组图片或特征变量[n,c,w,h],输出也为一组变量[n,c,w,h].变量类型为tenso...原创 2020-04-26 09:28:38 · 13646 阅读 · 7 评论 -
浅谈torch.nn库和torch.nn.functional库(Pytorch)
浅谈torch.nn库和torch.nn.functional库这两个库很类似,都涵盖了神经网络的各层操作,只是用法有点不同,nn下是类实现,nn.functional下是函数实现。conv1d在nn下是一个类,一般继承nn.module通过定义forward()函数计算其值class Conv1d(_ConvNd): def __init__(self, in_chan...原创 2019-12-23 12:16:17 · 802 阅读 · 0 评论