torch当中所有的层
在 PyTorch 里,层(layer)其实就是各种 torch.nn.Module 的子类。种类非常多,可以分为几大类:
1.卷积层
1.1. 1D/2D/3D卷积
nn.Conv1d, nn.Conv2d, nn.Conv3d
1.2.转置卷积(反卷积/上采样卷积)
nn.ConvTranspose1d, nn.ConvTranspose2d, nn.ConvTranspose3d
1.3.稀疏卷积/量化卷积
(部分在拓展库里)
2.池化层
2.1.最大池化
nn.MaxPool1d, nn...
yangyongbiao.hashnode.dev1 min read