PyTorch 是一个流行的开源机器学习库,广泛用于深度学习和计算机视觉领域。以下是一些官方文档的精选内容:
快速开始
安装 PyTorch
- 首先,您需要安装 PyTorch。请参考安装指南进行操作。
编写第一个 PyTorch 程序
- 安装完成后,您可以开始编写您的第一个 PyTorch 程序。以下是一个简单的例子:
import torch
# 创建一个张量
x = torch.tensor([1.0, 2.0, 3.0])
# 计算张量的和
print(x.sum())
- 探索 PyTorch API
- PyTorch 提供了丰富的 API,您可以在API 文档中找到详细信息。
图片示例
以下是一个 PyTorch 张量的示例: