code: def __init__(self, input_features: int, output_features: int):
      super().__init__()
      self.weight = nn.Parameter(
          torch.randn(input_features, output_features)
      )
  
file_path: opacus/opacus/tests/utils.py
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new MatmulModule object from inherited base classes: nn with example: obj = MatmulModule(input_features, output_features)