2. PyTorch Explained - Python Deep Learning Neural Network API
对于Python来说,最流行的科学计算包是numpy,它是n维数组的转换包,而Pytorch是一个张量库,它非常密切的反应了numpy的多维数组功能,它与numpy具有高度的互操作性。
它是一个很年轻的框架,16年才刚刚提出来
这个是我们用Pytorch主要调用的包
torch包含其他包和张量库的顶级包
torch.nn和torch.autograd是Pytorch的主要工作包,torch.nn是关于神经网络的包,包含类和模块 比如前向传播所用到的函数,权重等。torch.autograd是优化神经网络权重所用到的导数计算。
torch,nn.function是功能接口,它可以让我们访问损失函数,激活函数和卷积运算等。
torch.optim是优化的包,像SGD,Adam等
torch.utils是一个子包,它包括像数据集和数据加载器这样的实用程序类,使数据预处理更加容易。
torchvision可以让我们访问数据集,pytorch自带的数据集
另外,Pytorch在计算的时候用一个被称为动态计算图的计算图,随着操作的进行,图形是动态生成的