nn.Conv2d(in_channels, out_channels, kernel_size) # 卷积层。model.load_state_dict(torch.load(path)) # 加载权重。nn.Embedding(vocab_size, embedding_dim) # 词嵌入。torch.save(model.state_dict(), path) # 保存模型权重。nn.Li