class_docstr: Implementation of a Shallow model
  
  :param img_size (int): Dimension of input image
  :param hidden_size (int): Hidden layer dimension
  :param num_classes (int): Number of classes for classification
code: def forward(self, x):
      x = torch.flatten(x, 1)
      x = self.fc1(x)
      x = F.relu(x)
      x = self.fc2(x)
      x = F.relu(x)
      out = self.fc3(x)
  
      return out
  
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 2 for calling the Shallow obj's underlying member method code with example usage: obj.forward(x) and returns: out