code: def forward(self, x):
      x = self.conv1(x)
      x = self.conv2(x)  # (batch,32,7,7)
      x = x.view(x.size(0), -1)  # (batch,32*7*7)
      # Add drop
      # x = F.dropout(x, training=self.training)
      output = self.out(x)
      return output
  
file_path: AdvBox/DataPoison/poison_mnist_pytorch.py
imports: from __future__ import division
  from __future__ import print_function
  from builtins import range
  from past.utils import old_div
  import torch
  import torchvision
  import torch.nn
  import torch.nn.functional
  import torch.optim
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: output