code: def _make_stage(features, size):
      prior = nn.AdaptiveAvgPool2d(output_size=(size, size))
      conv = nn.Conv2d(features, features, kernel_size=1, bias=False)
      return nn.Sequential(prior, conv)
  
file_path: pywick/pywick/models/segmentation/lexpsp.py
imports: from .lex_extractors import *
label_desc: Use image node_id 2 for calling the PSPModule obj's underlying member method code with example usage: obj._make_stage(features, size) and returns: nn