10 lines
269 B
Python
10 lines
269 B
Python
|
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)
|