class_docstr: Position attention module
code: def __init__(self, in_channels, **kwargs):
      super(_PositionAttentionModule, self).__init__()
      self.conv_b = nn.Conv2d(in_channels, in_channels // 8, 1)
      self.conv_c = nn.Conv2d(in_channels, in_channels // 8, 1)
      self.conv_d = nn.Conv2d(in_channels, in_channels, 1)
      self.alpha = nn.Parameter(torch.zeros(1))
      self.softmax = nn.Softmax(dim=-1)
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from pywick.models.segmentation.da_basenets.segbase import SegBaseModel
label_desc: Use image node_id 1 to create a new _PositionAttentionModule object from inherited base classes: nn with example: obj = _PositionAttentionModule(in_channels)