code: def conv3x3(
      in_planes: int,
      out_planes: int,
      stride: Union[int, Sequence[int]] = 1,
  ):
      return Conv2d(
          in_planes,
          out_planes,
          kernel_size=3,
          stride=stride,
          padding=1,
          bias=False,
      )
  
imports: from typing import Union, Sequence, Callable
  import torch
  from torch.nn import Module, Sequential, BatchNorm2d, Conv2d, ReLU, ConstantPad3d, Identity, AdaptiveAvgPool2d, Linear
  from torch import Tensor
  from torch.nn.init import zeros_, kaiming_normal_
  from torch.nn.modules.flatten import Flatten
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: conv3x3(in_planes, out_planes, stride) and returns: Conv2d