1.torch.nn模块

torch.nn模块包含着torch已经准备好的层,方便使用者调用构建网络。以下内容介绍,卷积层、池化层、激活函数层、循环层、全连接层的相关使用方法。

824afbae3e2f69fb1d46413ec2b40b8.jpg

目标:完成Swin Transformer的代码编写。