code: def __init__(
      self,
      patch_size=16,
      stride=16,
      padding=0,
      in_chans=3,
      embed_dim=768,
      norm_layer=None,
  ):
      super().__init__()
      patch_size = (patch_size, patch_size)
      stride = (stride, stride)
      padding = (padding, padding)
      self.proj = nn.Conv2d(
          in_chans,
          embed_dim,
          kernel_size=patch_size,
          stride=stride,
          padding=padding,
      )
      self.norm = norm_layer(embed_dim) if norm_layer else nn.Identity()
  
imports: from torch import nn
label_desc: Use image node_id 1 to create a new PatchEmbed object from inherited base classes: nn with example: obj = PatchEmbed(patch_size, stride, padding, in_chans, embed_dim, norm_layer)