code: def __init__(
      self, in_features: int, out_features: int, hidden_dim: int = 8
  ):
      super().__init__()
      self.fc = nn.Linear(in_features, hidden_dim)
      self.extra_param = nn.Parameter(
          torch.randn(hidden_dim, out_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 LinearWithExtraParam object from inherited base classes: nn with example: obj = LinearWithExtraParam(in_features, out_features, hidden_dim)