pelican.layers.Aggregator2to2

class pelican.layers.Aggregator2to2(in_channels, out_channels, **kwargs)[source]

Bases: GeneralAggregator

Aggregator from edges (rank 2) to edges (rank 2).

Parameters:

  • aggregator (callable) – Aggregation function to use.

  • in_rank (int) – Input rank (0 for graph, 1 for nodes, 2 for edges).

  • out_rank (int) – Output rank (0 for graph, 1 for nodes, 2 for edges).

  • in_channels (int) – Number of input channels.

  • out_channels (int) – Number of output channels.

  • map_multipliers (bool) – Whether to use learnable multipliers for each aggregation map, by default True.

  • factorize (bool) – Whether to use factorized coefficients, by default False. Factorization reduces the number of parameters.

  • aggr (str) – Aggregation method to use (‘mean’, ‘sum’, ‘prod’, ‘amin’, ‘amax’), by default ‘mean’.