欢迎来到 PyTorch 快速入门教程的第一步!在这里,我们将带你了解如何使用 PyTorch 进行简单的深度学习任务。

安装 PyTorch

首先,你需要安装 PyTorch。你可以通过访问 PyTorch 官方网站 来获取详细的安装指南。

创建第一个 PyTorch 程序

让我们创建一个简单的 PyTorch 程序,用于加权和激活函数。

import torch

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

# 定义权重
weights = torch.tensor([0.5, 0.5, 0.5])

# 乘以权重
weighted = torch.mul(x, weights)

# 激活函数(例如 ReLU)
activated = torch.relu(weighted)

# 打印结果
print(activated)

深入学习

如果你想要更深入地了解 PyTorch,我们推荐阅读 PyTorch 教程

相关资源

PyTorch Logo