class_docstr: Image to Patch Embedding
      
class_docstr_tok: ["Image","to","Patch","Embedding"]
code: def forward(self, x):
      B, C, H, W = x.shape
  
      x = self.proj(x).flatten(2).transpose(1, 2)  # B Ph*Pw C
      if self.norm is not None:
          x = self.norm(x)
      return x
  
imports: import os
  import logging
  import torch
  import torch.nn
  import torch.nn.functional
  from functools import partial
  import torch.distributed
  from torch.nn.init import trunc_normal_
  from megatron.model.transformer import DropPath
  from megatron import get_args
  from megatron.model import LayerNorm
  import numpy
  from math import sqrt
label_desc: Use image node_id 2 for calling the PatchEmbed obj's underlying member method code with example usage: obj.forward(x) and returns: x