@@ -1,10 +0,0 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user