code: def forward(self, x):
      x = self.fc1(x)
      x = self.relu1(x)
      x = self.fc2(x)
      x = self.relu2(x)
      x = self.fc3(x)
      return x
  
file_path: pytorch/functorch/examples/maml_regression/evjang_transforms_module.py
imports: import math
  import matplotlib
  import matplotlib.pyplot
  import numpy
  import torch
  from functorch import grad, make_functional, vmap
  from torch import nn
  from torch.nn import functional
label_desc: Use image node_id 2 for calling the ThreeLayerNet obj's underlying member method code with example usage: obj.forward(x) and returns: x