code: def __init__(self, d_embed, d_proj):
      super().__init__()
      self.linear = nn.Linear(d_embed, d_proj)
  
imports: import unittest
  import torch
  import torch.nn.functional
  import torchvision
  import nni.nas.nn.pytorch.layers
  from nni.nas.nn.pytorch import ParametrizedModule
  from .convert_mixin import ConvertMixin, ConvertWithShapeMixin
label_desc: Use image node_id 1 to create a new Linear object from inherited base classes: ParametrizedModule with example: obj = Linear(d_embed, d_proj)