pelican.primitives.aggregate_0to2

pelican.primitives.aggregate_0to2(graph, edge_index, batch, **kwargs)[source]

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

Parameters:

  • graph (torch.Tensor) – Graph-level features of shape (G, C).

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

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

  • **kwargs

Returns:

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

Return type:

torch.Tensor