code: def _register_att_map(self, g, enc_ids, dec_ids):
      self.att_weight_map = [
          get_attention_map(g, enc_ids, enc_ids, self.h),
          get_attention_map(g, enc_ids, dec_ids, self.h),
          get_attention_map(g, dec_ids, dec_ids, self.h),
      ]
  
imports: from .config import *
  from .act import *
  from .attention import *
  from .viz import *
  from .layers import *
  from .functions import *
  from .embedding import *
  import threading
  import dgl.function
  import torch
  import torch.nn.init
label_desc: Use image node_id 6 for calling the Transformer obj's underlying member method code with example usage: obj._register_att_map(g, enc_ids, dec_ids) without return types