派生pytorch网络的结构

import torch
import torch.nn as nn
import torch.nn.functional as F

class MyNetwork(nn.Module):
    def __init__(self, input_size, hidden_size, output_size):
        super(MyNetwork, self).__init__()
        self.fc1 = nn.Linear(input_size, hidden_size)
        self.fc2 = nn.Linear(hidden_size, output_size)

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

input_size = 10
hidden_size = 20
output_size = 5
net = MyNetwork(input_size, hidden_size, output_size)

input_data = torch.randn(32, input_size)  # 随机生成输入数据，大小为[batch_size, input_size]
output = net(input_data)  # 前向传播计算输出