首先我们先搭建一个模型: import torch from torch import nn import torch.nn.functional as F class Net(nn.Module): def __init__(self): super(Net