import torch


class PolyModel(torch.nn.Module):
    def __init__(self,input_size,output_size):
        super().__init__()
        self.linear = torch.nn.Linear(input_size, output_size)  # 输入n维，输出m维

    def forward(self, x):
        return self.linear(x)