EdgeWeighting
- class EdgeWeighting(**kwargs)[source]
Bases:
torch.nn.modules.module.Module
Base class for edge weightings.
Initializes internal Module state, shared by both nn.Module and ScriptModule.
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