If each weight has a weight, we can regard the weight as the information. I want to pass messages group by the information. Does DGL have the API?
pass messages example:
def gcn_msg(edge):
msg = edge.src['h'] * edge.src['norm']
return {'m': msg}
By the way, does DGL pass messages on the whole edges at once? I want to batch the edges when passing the message.