code: def __init__(self, N: int, M: int) -> None:
      super().__init__()
      self.weight = torch.nn.Parameter(torch.ones(N, M))
  
file_path: torchrec/torchrec/inference/tests/test_modules.py
imports: import torch
label_desc: Use image node_id 1 to create a new Simple object from inherited base classes: torch.nn with example: obj = Simple(N, M)