PyTorch 是一个开源的机器学习库，由 Facebook 的 AI 研究团队开发，用于应用在深度学习领域。它提供了灵活且易于使用的接口，让研究人员和开发者能够轻松地构建和训练复杂的神经网络。
安装 PyTorch
首先，您需要安装 PyTorch。您可以从 PyTorch 官网 获取详细的安装指南。
快速开始
以下是一个简单的 PyTorch 程序示例，用于计算矩阵的乘积：
import torch
# 创建两个张量
x = torch.tensor([1, 2, 3], dtype=torch.float)
y = torch.tensor([4, 5, 6], dtype=torch.float)
# 计算矩阵乘积
z = x * y
print(z)
PyTorch 社区
PyTorch 拥有一个庞大的社区，您可以在 PyTorch 论坛 上找到各种资源和帮助。
相关资源
PyTorch Logo