def METHOD_NAME(self):
    alpha, normalizer = self.softmax(self.g.edata['a'], self.g)
    self.g.ndata['z'] = normalizer
    if self.training and self.dropout > 0:
        alpha = F.dropout(alpha, p=self.dropout, training=True)
    self.g.edata['a'] = alpha