首先,我们需要在 PyTorch 中导入所需的库。 import torch import torch.nn as nn import torch.optim as optim import matplotlib.pyplot as plt