Hello everyone,
Like in the first section of https://docs.dgl.ai/guide/message.html , I would like to update both edges and nodes features per step.
I was wondering how I could do it using DGL:
def edge_message_func(edges):
return {'me': phi(edges.src['h'], edges.dst['h'], edges.data['e'])}
def reduce_func(nodes):
return {'h': rho(nodes.mailbox['me'])}
def apply_func(nodes):
return {'h': psi(nodes.data['h']}
Does that make sense?
Thanks.