在 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 readNo responses yet.