AttentionEdgeWeighting¶
- class AttentionEdgeWeighting(message_dim, num_heads=8, dropout=0.1)[source]¶
Bases:
EdgeWeighting
Message weighting by attention.
Initialize the module.
- Parameters:
- Raises:
ValueError – If
message_dim
is not divisible bynum_heads
Attributes Summary
whether the edge weighting needs access to the message
Methods Summary
forward
(source, target[, message, x_e])Compute edge weights.
Attributes Documentation
Methods Documentation