code: def build_output_layer(self):
      return RelGraphConv(
          self.h_dim,
          self.out_dim,
          self.num_rels,
          "basis",
          self.num_bases,
          activation=partial(tf.nn.softmax, axis=1),
          self_loop=self.use_self_loop,
      )
  
file_path: dgl/examples/tensorflow/rgcn/entity_classify.py
imports: import argparse
  import time
  from functools import partial
  import dgl
  import numpy
  import tensorflow
  from dgl.data.rdf import AIFBDataset, AMDataset, BGSDataset, MUTAGDataset
  from dgl.nn.tensorflow import RelGraphConv
  from model import BaseRGCN
  from tensorflow.keras import layers
label_desc: Use image node_id 4 for calling the EntityClassify obj's underlying member method code with example usage: obj.build_output_layer() and returns: RelGraphConv