code: def conv_1xnxn(inp, oup, kernel_size=3, stride=3, groups=1):
      return nn.Conv3d(
          inp,
          oup,
          (1, kernel_size, kernel_size),
          (1, stride, stride),
          (0, 0, 0),
          groups=groups,
      )
  
imports: from collections import OrderedDict
  import torch
  from torch import nn
  from torch.utils import checkpoint
  from functools import partial
  from towhee.models.utils.general_utils import to_2tuple
  from towhee.models.utils.weight_init import trunc_normal_
  from towhee.models.layers.droppath import DropPath
  from towhee.models.layers.mlp import Mlp
  from towhee.models.layers.attention import MultiHeadAttention
  from towhee.models.uniformer.config import _C
  import os
label_desc: Use image node_id 2 for calling a global function with example usage: conv_1xnxn(inp, oup, kernel_size, stride, groups) and returns: nn