class_docstr: PatchEmbed.
class_docstr_tok: ["PatchEmbed","."]
code: def forward(self, x, keep_spatial=False):
      x = self.proj(x)
      if keep_spatial:
          return x, x.shape
      # B C (T) H W -> B (T)HW C
      return x.flatten(2).transpose(1, 2), x.shape
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the PatchEmbed obj's underlying member method code with example usage: obj.forward(x, keep_spatial) and returns: x,  x, x,  x