PyTorch 是一个流行的开源机器学习库,广泛用于深度学习和计算机视觉领域。以下是一些官方文档的精选内容:

快速开始

  1. 安装 PyTorch

    • 首先,您需要安装 PyTorch。请参考安装指南进行操作。
  2. 编写第一个 PyTorch 程序

    • 安装完成后,您可以开始编写您的第一个 PyTorch 程序。以下是一个简单的例子:
import torch

# 创建一个张量
x = torch.tensor([1.0, 2.0, 3.0])

# 计算张量的和
print(x.sum())
  1. 探索 PyTorch API
    • PyTorch 提供了丰富的 API,您可以在API 文档中找到详细信息。

图片示例

以下是一个 PyTorch 张量的示例:

tensor

返回首页