torch基本功能介绍

torch的基本功能

torch:张量的相关运算，eg:创建、索引、切片、连续、转置、加减乘除等相关运算。

torch.nn:包含搭建网络层的模块（modules)和一系列的loss函数。eg.全连接、卷积、池化、BN分批处理、dropout、CrossEntropyLoss、MSLoss等。

torch.autograd:提供Tensor所有操作的自动求导方法。

torch.nn.functional:常用的激活函数relu、leaky_relu、sigmoid等。

torch.optim:各种参数优化方法，例如SGD、AdaGrad、RMSProp、Adam等。

torch.nn.init:可以用它更改nn.Module的默认参数初始化方式。

torch.utils.data:用于加载数据。

Torchvision基本功能

torchvision.datasets:常用数据集,MNIST、COCO、CIFAR10等。

torchvision.models:常用模型AlextNet、VGG、ResNet、DenseNet等。

torchvision.transforms:图片相关处理。裁剪、尺寸缩放、归一化等。

torchvision.utils:将给定的Tensor保存成image文件。

ps.因为我也是这个领域的学徒，有什么问题请指出来，一起共同进步,谢谢您！