深度学习之pytorch 中 torch.nn介绍
pytorch 中必用的包就是 torch.nn,torch.nn 中按照功能分,主要如下有几类:1. Layers(层):包括全连接层、卷积层、池化层等。2. Activation Functions(激活函数):包括ReLU、Sigmoid、Tanh等。3. Loss Functions(损失函数):包括交叉熵损失、均方误差等。4. Optimizers(优化器):包括SGD、Adam、RMSprop等。
原创
2024-02-20 16:21:05 ·
1535 阅读 ·
2 评论