code: def forward(self, x):
      x = self.pool(F.relu(self.conv1(x)))
      x = self.pool(F.relu(self.conv2(x)))
      x = x.view(-1, 16 * 5 * 5)
      x = F.relu(self.fc1(x))
      x = F.relu(self.fc2(x))
      x = self.fc3(x)
      return x
  
imports: from __future__ import print_function
  import os
  from os.path import exists
  from os import makedirs
  import argparse
  import matplotlib.pyplot
  import numpy
  import torch
  import torchvision
  import torchvision.transforms
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from bigdl.dllib.utils.log4Error import invalidInputError
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.pytorch import Estimator
  from bigdl.orca.learn.metrics import Accuracy
  from bigdl.orca.learn.trigger import EveryEpoch
label_desc: Use image node_id 2 for calling the Net obj's underlying member method code with example usage: obj.forward(x) and returns: x