in source/sagemaker/sagemaker_graph_fraud_detection/dgl_fraud_detection/model/mxnet.py [0:0]
def __init__(self, in_size, out_size, etypes):
super(HeteroRGCNLayer, self).__init__()
# W_r for each relation
with self.name_scope():
self.weight = {name: gluon.nn.Dense(out_size, use_bias=False) for name in etypes}
for child in self.weight.values():
self.register_child(child)