code: def __init__(self, in_dim=128):
      super().__init__()
      self.linear1 = nn.Linear(in_dim, 128)
      self.relu1 = nn.ReLU()
      self.linear2 = nn.Linear(128, 128)
      self.relu2 = nn.ReLU()
      self.linear3 = nn.Linear(128, 1)
  
imports: import collections
  import itertools
  import logging
  import unittest
  import crypten
  import torch
  import torch.nn
  import torch.nn.functional
  from crypten.config import cfg
  from crypten.cuda import CUDALongTensor
  from test.multiprocess_test_case import get_random_test_tensor, MultiProcessTestCase
  from test.test_mpc import TestMPC
label_desc: Use image node_id 1 to create a new MLP object from inherited base classes: nn with example: obj = MLP(in_dim)