code: def forward(self, x):
      x = self.fc1(x)
      x = self.sub_fc(x, self.fc1.weight)
      x = self.fc1(x)
      x = self.fc2(x)
      return x
  
file_path: ColossalAI/tests/kit/model_zoo/custom/nested_model.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  from ..registry import model_zoo
  from .base import CheckpointModule
label_desc: Use image node_id 2 for calling the NestedNet obj's underlying member method code with example usage: obj.forward(x) and returns: x