code: def pool2d(_w_in, k, *, stride=1):
      """Helper for building a pool2d layer."""
      assert (
          k % 2 == 1
      ), "Only odd size kernels supported to avoid padding issues."
      return nn.MaxPool2d(k, stride=stride, padding=(k - 1) // 2)
  
imports: import numpy
  import torch
  import torch.nn
  from pycls.core.config import cfg
  from torch.nn import Module
label_desc: Use image node_id 4 for calling a global function with example usage: pool2d(_w_in, k) and returns: nn