torch

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文件。