class_docstr: Image to Patch Embedding
      
code: def forward(self, x) -> Tuple[torch.Tensor, Size_]:
      B, C, H, W = x.shape
  
      x = self.proj(x).flatten(2).transpose(1, 2)
      x = self.norm(x)
      out_size = (H // self.patch_size[0], W // self.patch_size[1])
  
      return x, out_size
  
imports: import math
  from functools import partial
  from typing import Tuple
  import torch
  import torch.nn
  import torch.nn.functional
  from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
  from timm.layers import Mlp, DropPath, to_2tuple, trunc_normal_, use_fused_attn
  from ._builder import build_model_with_cfg
  from ._features_fx import register_notrace_module
  from ._registry import register_model, generate_default_cfgs
  from .vision_transformer import Attention
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,  out_size