code: def __init__(self, in_size, out_size, device=None, hidden_size=64):
      super().__init__()
      self._linear0 = TorchLinearModule(in_size, hidden_size)
      self._linear1 = TorchLinearModule(hidden_size, hidden_size)
      self._linear2 = TorchLinearModule(hidden_size, out_size)
  
imports: import pytest
  from types import SimpleNamespace
  from typing import Sequence
  import ivy
label_desc: Use image node_id 1 to create a new TorchModule object from inherited base classes: nn with example: obj = TorchModule(in_size, out_size, device, hidden_size)