【Pytorch实用教程】Pytorch的torch.nn模块中都有哪些函数

在PyTorch的torch.nn模块中，有许多用于构建和训练神经网络的函数和类。以下是一些主要的函数和类分类：

1. 神经网络层 (Layers)

• nn.Linear：全连接层
• nn.Conv2d：二维卷积层
• nn.Conv3d：三维卷积层
• nn.ConvTranspose2d：二维反卷积层
• nn.ConvTranspose3d：三维反卷积层
• nn.BatchNorm1d：一维批量归一化层
• nn.BatchNorm2d：二维批量归一化层
• nn.BatchNorm3d：三维批量归一化层
• nn.RNN：简单循环神经网络层
• nn.LSTM：长短期记忆网络层
• nn.GRU：门控循环单元层
• nn.Embedding：嵌入层
• nn.Dropout：Dropout层
• nn.D