Pytorch库的基本架构介绍
今天我就将PyTorch
常用的模块做一个总结梳理。
首先要说明的是PyTorch
这是torch
的Python
版本,所以导入的是torch
而不是Pytorch
:
import torch
1 运行基础
-
torch.tensor
:基础数据结构 -
torch.autograd
:自动微分模块
2 torch.utils
支持神经网络相关的数据预处理。
-
数据导入与处理
-
utils.data
-
utils.datasets
-
-
utils.tensorboard
:训练结果的可视化 -
utils.model_zoo
:预训练模型
3 torch.nn
构建神经网络结构的基本元素。
-
nn.Module
:神经网络的各种结构“层” -
nn.functional
:神经网络的损失函数与激活函数
4 torch.optim
神经网络的算法优化模块,封装着各类优化器。