> 文章列表 > 深度学习-PyTorch 基础

深度学习-PyTorch 基础

深度学习-PyTorch 基础

深度学习-PyTorch 基础

介绍

PyTorch 是由 Facebook 主导开发的深度学习框架,因其高效的计算过程以及良好的易用性被诸多大公司和科研人员所喜爱。本次实验中,我们将学习 PyTorch 的基础语法,了解 Autograd 自动求导机制,并最终利用 PyTorch 构建可用于图像分类任务的人工神经网络。

知识点

  • PyTorch 基础语法
  • Autograd 自动求导
  • 神经网络分类器

PyTorch 基础语法

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-8ZoZQnO3-1681797873110)(null)]

* PyTorch* 是 Facebook 主导开发的,基于 Python 的科学计算包,其主要有以下两个特点:

  • 比 NumPy 更灵活,可以使用 GPU 的强大计算能力。
  • 开源高效的深度学习研究平台。

接下来,我们了解 PyTorch 的基础概念和语法。

张量

Tensors(张量)与 NumPy 中的 Ndarrays 多维数组类似,但是在 PyTorch 中 Tensors 可以使用 GPU 进行计算。

下面,我们创建一个 5×35×3 矩阵。使用