class_docstr: The patch embedding layer at stage 0.
  
  Args:
      num (`int`): number of repeated sets of layers
class_docstr_tok: ["The","patch","embedding","layer","at","stage","0",".","Args",":","num","(","`","int","`",")",":","number","of","repeated","sets","of","layers"]
code: def forward(self, x):
      x = self.conv(x)
      _, _, h, w = x.shape
      x = x.flatten(2).transpose(1, 2)
      x = self.norm(x)
      return x, h, w
  
imports: import torch
  from torch import nn
  from towhee.models.layers.patch_embed2d import PatchEmbed2D
  from towhee.models.layers.mlp import Mlp
  from towhee.models.layers.droppath import DropPath
  from towhee.models.utils.init_vit_weights import init_vit_weights
label_desc: Use image node_id 2 for calling the HeadPatchEmbed obj's underlying member method code with example usage: obj.forward(x) and returns: x,  h,  w