code: def normalize_adj(self, adj):
      adj.fill_diagonal_(1)
      # normalize adj with A = D^{-1/2} @ A @ D^{-1/2}
      D_norm = torch.diag(torch.pow(adj.sum(1), -0.5)).cuda()
      adj = D_norm @ adj @ D_norm
      return adj
  
file_path: DIG/dig/fairgraph/method/Graphair/aug_module.py
imports: import torch
  import torch.nn
  import torch.nn.functional
  import pyro
  from .GCN import GCN_Body
label_desc: Use image node_id 3 for calling the aug_module obj's underlying member method code with example usage: obj.normalize_adj(adj) and returns: adj