class_docstr: 2D Image to Patch Embedding
class_docstr_tok: ["2D","Image","to","Patch","Embedding"]
code: def __init__(
      self, patch_size=4, in_c=3, embed_dim=96, norm_layer=None
  ):
      super().__init__()
      patch_size = (patch_size, patch_size)
      self.patch_size = patch_size
      self.in_chans = in_c
      self.embed_dim = embed_dim
      self.proj = nn.Conv2d(
          in_c, embed_dim, kernel_size=patch_size, stride=patch_size
      )
      self.norm = norm_layer(embed_dim) if norm_layer else nn.Identity()
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  import torch.utils.checkpoint
  import numpy
  from typing import Optional
label_desc: Use image node_id 1 to create a new PatchEmbed object from inherited base classes: nn with example: obj = PatchEmbed(patch_size, in_c, embed_dim, norm_layer)