HelpingHand.py

#Import Torch.nn
from torch import nn

#Declaring Class
class Network(nn.Module):
  def __init__():
    super(type(self), self).init() 
    self.hidden = nn.Linear(784, 256)
    self.output = nn.Linear(256, 10)
    self.sigmoid = nn.Sigmoid()
    self.softmax = nn.Softmax(dim=1)

  def forward(self, x):
    x = self.hidden(x)
    x = self.sigmoid(x)
    x = self.output(x)
    x = self.softmax(x)

    return x
  
# Initialing the object
model = Network()