class_docstr: 2D Image to Patch Embedding
class_docstr_tok: ["2D","Image","to","Patch","Embedding"]
code: def call(self, inputs, **kwargs):
      B, H, W, C = inputs.shape
      assert (
          H == self.img_size[0] and W == self.img_size[1]
      ), f"Input image size ({H}*{W}) doesn't match model ({self.img_size[0]}*{self.img_size[1]})."
      x = self.proj(inputs)
      # [B, H, W, C] -> [B, H*W, C]
      x = tf.reshape(x, [B, self.num_patches, self.embed_dim])
      return x
  
imports: import tensorflow
  from tensorflow.keras import Model, layers, initializers
  import numpy
label_desc: Use image node_id 2 for calling the PatchEmbed obj's underlying member method code with example usage: obj.call(inputs) and returns: x