class_docstr: PatchEmbed.
class_docstr_tok: ["PatchEmbed","."]
code: def forward(self, x):
      x = self.proj(x)
      if self.conv_2d:
          self.output_h, self.output_w = x.shape[-2:]
      else:
          self.output_t, self.output_h, self.output_w = x.shape[-3:]
      # B C (T) H W -> B (T)HW C
      return x.flatten(2).transpose(1, 2)
  
imports: import math
  from functools import partial
  from typing import Tuple, List
  import numpy
  import torch
  from torch import nn
  from torch.nn.init import trunc_normal_
  from einops import rearrange
  from towhee.models.layers.layers_with_relprop import GELU, Linear, Dropout, Conv2d, Conv3d, Softmax, Sigmoid, LayerNorm, Einsum, Add, Clone, MaxPool3d, IndexSelect
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