class FeedForward_NN(nn.Module):
def init(self,input_dim,output_dim):
super(FeedForward_NN,self).init()
    self.fc0 = nn.Linear(input_dim, output_dim)
    
def forward(self,x):

    out = self.fc0(x)
    return out