class_docstr: Image to Patch Embedding.
class_docstr_tok: ["Image","to","Patch","Embedding","."]
code: def forward(self, x: Tensor) -> Tensor:
      x = self.proj(x)
      # B C H W -> B H W C
      x = x.permute(0, 2, 3, 1)
      return x
  
imports: from __future__ import annotations
  from typing import Optional
  import torch
  import torch.nn.functional
  from torch import nn
  from kornia.contrib.models.common import LayerNorm2d, window_partition, window_unpartition
  from kornia.contrib.models.sam.architecture.common import MLPBlock
  from kornia.core import Module, Tensor, zeros
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