pelican.primitives.aggregate_2to2

pelican.primitives.aggregate_2to2(edges, edge_index, batch, reduce='mean', **kwargs)[source]

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

Parameters:

  • edges (torch.Tensor) – Edge-level features of shape (E, C).

  • edge_index (torch.Tensor) – Edge index tensor of shape (2, E).

  • batch (torch.Tensor) – Batch tensor of shape (N).

  • reduce (str, optional) – Reduction method to use (‘mean’, ‘sum’, ‘prod’, ‘amin’, ‘amax’), by default ‘mean’.

  • **kwargs

Returns:

ops – Aggregated features of shape (E, C, 15).

Return type:

torch.Tensor