【pytorch】张量求导3

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

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

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