class_docstr: 2D Image to Patch Embedding
class_docstr_tok: ["2D","Image","to","Patch","Embedding"]
code: def __init__(self, patch_size=4, embed_dim=96, norm_layer=None):
      super(PatchEmbed, self).__init__()
      self.embed_dim = embed_dim
      self.patch_size = (patch_size, patch_size)
      self.norm = (
          norm_layer(epsilon=1e-6, name="norm")
          if norm_layer
          else layers.Activation("linear")
      )
  
      self.proj = layers.Conv2D(
          filters=embed_dim,
          kernel_size=patch_size,
          strides=patch_size,
          padding="SAME",
          kernel_initializer=initializers.LecunNormal(),
          bias_initializer=initializers.Zeros(),
          name="proj",
      )
  
imports: import tensorflow
  from tensorflow.keras import Model, layers, initializers
  import numpy
label_desc: Use image node_id 1 to create a new PatchEmbed object from inherited base classes: layers with example: obj = PatchEmbed(patch_size, embed_dim, norm_layer)