PyTorch:如何编写只返回权重的神经网络？

import torch
import torch.nn as nn

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(10, 20)
        self.fc2 = nn.Linear(20, 1)

    def forward(self, x):
        x = torch.relu(self.fc1(x))
        x = self.fc2(x)
        return x

# 创建神经网络模型
model = Net()

# 获取模型的权重
weights = model.state_dict()

# 打印权重
for name, param in weights.items():
    print(name, param)