欢迎来到 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