class_docstr: Auxiliary head in 2/3 place of network to let the gradient flow well 
code: def forward(self, x):
      out = self.net(x)
      out = out.view(out.size(0), -1)  # flatten
      logits = self.linear(out)
      return logits
  
imports: from collections import OrderedDict
  import torch
  import torch.nn
  import ops
  from nni.retiarii.nn.pytorch import LayerChoice, InputChoice
label_desc: Use image node_id 2 for calling the AuxiliaryHead obj's underlying member method code with example usage: obj.forward(x) and returns: logits