code: def forward(self, x):
      feat = self.conv(x)
      atten = F.avg_pool2d(feat, feat.size()[2:])
      atten = self.conv_atten(atten)
      atten = self.bn_atten(atten)
      atten = self.sigmoid_atten(atten)
      out = torch.mul(feat, atten)
      return out
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from ParsingResnet import Resnet18
label_desc: Use image node_id 2 for calling the AttentionRefinementModule obj's underlying member method code with example usage: obj.forward(x) and returns: out