class_docstr: Image to Patch Embedding.
code: def forward(self, x: torch.Tensor) -> torch.Tensor:
      x = self.proj(x)
      # B C H W -> B H W C
      x = x.permute(0, 2, 3, 1)
      return x
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from typing import Optional, Tuple, Type
  from .common import LayerNorm2d, MLPBlock
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