PyTorch包括一个特殊功能,可以创建和实现神经网络。在本章中,我们将创建一个简单的神经网络,其中包含一个隐藏层和一个输出单元。
我们将使用以下步骤来使用PyTorch实现第一个神经网络:
**步骤 1**
首先,我们需要导入PyTorch库,如下所示:
import torch
import torch.nn as nn
**步骤 2**
定义所有的层和批量大小,以开始执行神经网络,如下所示:
# 定义输入大小,隐藏层大小,输出大小和批量大小
n_in, n_h, n_out, batch_size = 10, 5, 1, 10
**步骤 3**
由于神经网络包括将输入数据组合以获取相应输出数据的过程,我们将按照以下方式执行相同的过程:
# 创建虚拟输入和目标张量(数据)
x = torch.randn(batch_size, n_in)
y = torch.tensor([[1.0], [0.0], [0.0], [1.0], [1.0], [1.0], [0.0], [0.0], [1.0], [1.0]])
*