☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜
PyTorch是一个基于Python的深度学习框架，用于构建各种神经网络。本文将展示如何使用PyTorch构建简单的神经网络，并提供代码示例。
首先，我们需要安装PyTorch。可以通过以下命令在命令行中安装：
pip install torch
接下来，我们将使用PyTorch构建一个简单的全连接神经网络，用于二元分类任务。这个神经网络将有两个隐藏层，每个隐藏层有10个神经元。我们将使用sigmoid激活函数和交叉熵损失函数。
以下是完整的代码：
import torch
import torch.nn as nn
import torch.optim as optim
# 定义神经网络模型
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
self.fc1 = nn.Linear(2, 10) # 第一个隐藏层
self.fc2 = nn.Linear(10, 10) # 第二个隐藏层
self.fc3 = nn.Linear(10, 1) # 输出层
def forward(self, x):
x = torch.sigmoid(self.fc1(x))
x = torch.sigmoid(self.fc2(x))
x = torch.sigmoid(self.fc3(x))
return x
# 创建数据集
X = torch.tensor([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=torch.float32)
y = torch.tensor([[0], [1], [1], [0]], dtype=torch.float32)
# 创建神经网络实例
net = Net()
# 定义损失函数和优化器
criterion = nn.BCELoss()
optimizer = optim.SGD(net.parameters(), lr=0.1)
# 训练神经网络
for epoch in range(10000):
optimizer.zero_grad()
output = net(X)
loss = criterion(output, y)
loss.backward()
optimizer.step()
# 打印训练损失
if epoch % 1000 == 0:
print('Epoch {}: loss = {}'.format(epoch, loss.item()))
# 使用训练好的神经网络进行预测
with torch.no_grad():
output = net(X)
predicted = (output > 0.5).float()
print('Predicted: {}\n'.format(predicted))首先，我们定义了一个名为Net的类，它继承自nn.Module。这个类包含了神经网络的所有层。在这个例子中，我们定义了三个全连接层，其中前两个是隐藏层，最后一个是输出层。
在Net类中，除了定义了一个forward方法来描述神经网络的前向传播过程外，我们还使用了sigmoid激活函数将每个隐藏层的输出传递到下一层。
接下来，我们创建了一个包含四个样本的数据集，其中每个样本有两个特征。我们还定义了一个名为net的神经网络实例，并选择了BCELoss作为损失函数和SGD作为优化器。
然后，我们开始训练神经网络。在每个迭代中，我们首先将优化器的梯度清零，然后将数据集X传递到神经网络中，以获取输出。我们计算损失并进行反向传播，最后使用优化器更新网络参数。我们还打印了每1000个迭代的训练损失。
训练完成后，我们使用no_grad上下文管理器对数据集进行预测。我们将输出四个预测结果，并打印它们。
这是一个简单的例子，演示了如何使用PyTorch构建基本的神经网络。PyTorch提供了很多工具和函数，可以帮助我们更轻松地构建和训练神经网络。
