class_docstr: Aggregate the context features according to the initial
  predicted probability distribution.
  Employ the soft-weighted method to aggregate the context.
code: def __init__(self, cls_num=0, scale=1):
      super(SpatialGather_Module, self).__init__()
      self.cls_num = cls_num
      self.scale = scale
  
imports: import os
  import logging
  import numpy
  import torch
  import torch.nn
  import torch._utils
  import torch.nn.functional
  from .bn_helper import BatchNorm2d, BatchNorm2d_class, relu_inplace
label_desc: Use image node_id 1 to create a new SpatialGather_Module object from inherited base classes: nn with example: obj = SpatialGather_Module(cls_num, scale)