class_docstr: Graph Transformer Layer
class_docstr_tok: ["Graph","Transformer","Layer"]
code: def forward(self, A, h):
      h1 = h
      h = self.MHA(A, h)
      h = self.batchnorm1(h + h1)
  
      h2 = h
      h = self.FFN2(F.relu(self.FFN1(h)))
      h = h2 + h
  
      return self.batchnorm2(h)
  
imports: import dgl
  import dgl.nn
  import dgl.sparse
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from dgl.data import AsGraphPredDataset
  from dgl.dataloading import GraphDataLoader
  from ogb.graphproppred import collate_dgl, DglGraphPropPredDataset, Evaluator
  from ogb.graphproppred.mol_encoder import AtomEncoder
  from tqdm import tqdm
label_desc: Use image node_id 2 for calling the GTLayer obj's underlying member method code with example usage: obj.forward(A, h) and returns: self