再接上文，补一下作者未补完的矩阵运算的坑。

首先贴一下原作者的图，将其转化为如下代码：

import torch
import torch.nn as nn
import torch.optim as optim

# 定义一个简单的两层神经网络
class TwoLayerNet(nn.Module):
    def __init__(self):
        super(TwoLayerNet, self).__init__()
        self.fc1 = nn.Linear(2, 2)  # 第一层
        self.fc2 = nn.Linear(2, 2)  # 第二层

    def forward(self, x):
        x = self.fc1(x)  # 第一层的前向传播
        x = self.fc2(x