torch与torchvision

torch和torchvision是PyTorch深度学习框架的两个重要模块。

torch是PyTorch的核心库，提供了张量操作、自动求导、神经网络等基础功能。它类似于NumPy，但更适用于GPU加速的深度学习任务。torch可以用于构建和训练神经网络模型，并提供了丰富的各种函数和方法来支持模型的操作和优化。

torchvision是基于torch的一个图像处理库，提供了一系列用于处理图像数据集的工具和预训练模型。它包含了常用的数据集加载、数据转换、图像变换、模型定义和预训练模型等功能。torchvision可以帮助开发者快速搭建图像分类、目标检测、图像生成等任务的深度学习模型。