水跃鱼

原创 Pytorch常用工具箱

神经网络工具箱nn import torch.nn as nn 在nn中主要有两个重要模块:nn.Model、nn.functional，接着将分别介绍这两个模块。 nn.Model nn.Model是nn的一个核心数据结构，最常用的做法就是继承nn.Model，如 class Nets(nn.Model)，常用的全连接层、损失层、激活层、卷积层等都是nn.Model的子类，nn.Linear、nn.Conv2d等 nn.functional import torch.nn.functional as F