easygraph.nn.convs.common module#
- class easygraph.nn.convs.common.MLP(*args: Any, **kwargs: Any)[source]#
Bases:
Module
adapted from CUAI/CorrectAndSmooth
- class easygraph.nn.convs.common.MultiHeadWrapper(*args: Any, **kwargs: Any)[source]#
Bases:
Module
A wrapper to apply multiple heads to a given layer.
- Parameters:
num_heads (
int
) – The number of heads.readout (
bool
) – The readout method. Can be"mean"
,"max"
,"sum"
, or"concat"
.layer (
nn.Module
) – The layer to apply multiple heads.**kwargs – The keyword arguments for the layer.