class_docstr: PatchEmbed.
class_docstr_tok: ["PatchEmbed","."]
code: def __init__(
      self,
      dim_in=3,
      dim_out=768,
      kernel=(1, 16, 16),
      stride=(1, 4, 4),
      padding=(1, 7, 7),
      conv_2d=False,
  ):
      super().__init__()
      if conv_2d:
          conv = nn.Conv2d
      else:
          conv = nn.Conv3d
      self.proj = conv(
          dim_in,
          dim_out,
          kernel_size=kernel,
          stride=stride,
          padding=padding,
      )
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new PatchEmbed object from inherited base classes: nn with example: obj = PatchEmbed(dim_in, dim_out, kernel, stride, padding, conv_2d)