deephyper.keras.layers.MessagePasserNNM#
- class deephyper.keras.layers.MessagePasserNNM(*args: Any, **kwargs: Any)[source]#
Bases:
Layer
Message passing kernel.
- Parameters:
Methods
build
Apply the layer on input tensors.
- call(inputs, **kwargs)[source]#
Apply the layer on input tensors.
- Parameters:
inputs (list) – X (tensor): node feature tensor (batch size * # nodes * state dimension) A (tensor): edge pair tensor (batch size * # edges * 2), one is source ID, one is target ID E (tensor): edge feature tensor (batch size * # edges * # edge features) degree (tensor): node degree tensor for GCN attention (batch size * # edges)
- Returns:
results after edge network, attention and aggregation (batch size * # nodes * state dimension)
- Return type:
output (tensor)