torch的基本功能
torch:张量的相关运算,eg:创建、索引、切片、连续、转置、加减乘除等相关运算。
torch.nn:包含搭建网络层的模块(modules)和一系列的loss函数。eg.全连接、卷积、池化、BN分批处理、dropout、CrossEntropyLoss、MSLoss等。\n\n
torch.autograd:提供Tensor所有操作的自动求导方法。\n\n
torch.nn.functional:常用的激活函数relu、leaky_relu、sigmoid等。
\n\n
torch.optim:各种参数优化方法,例如SGD、AdaGrad、RMSProp、Adam等。\n\n
torch.nn.init:可以用它更改nn.Module的默认参数初始化方式。\n\n
torch.utils.data:用于加载数据。\n\n
Torchvision基本功能\ntorchvision.datasets:常用数据集,MNIST、COCO、CIFAR10等。\n\n
torchvision.models:常用模型AlextNet、VGG、ResNet、DenseNet等。\n\n
torchvision.transforms:图片相关处理。裁剪、尺寸缩放、归一化等。\n\n
torchvision.utils:将给定的Tensor保存成image文件。