code: def __init__(
      self,
      in_size,
      out_size,
      device=None,
      hidden_size=64,
      v=None,
      with_partial_v=False,
  ):
      self._linear0 = ivy.Linear(in_size, hidden_size, device=device)
      self._linear1 = ivy.Linear(
          hidden_size, hidden_size, device=device
      )
      self._linear2 = ivy.Linear(hidden_size, out_size, device=device)
      ivy.Module.__init__(
          self, device=device, v=v, with_partial_v=with_partial_v
      )
  
imports: import os
  from hypothesis import given, strategies
  import numpy
  import ivy
  from ivy.functional.ivy.gradients import _variable
  import ivy_tests.test_ivy.helpers
label_desc: Use image node_id 1 to create a new TrainableModule object from inherited base classes: ivy with example: obj = TrainableModule(in_size, out_size, device, hidden_size, v, with_partial_v)