pylibcugraphops.operators.agg_dmpnn_e2e_fwd#
- pylibcugraphops.operators.agg_dmpnn_e2e_fwd = <nanobind.nb_func object>#
- Computes the forward pass for D-MPNN-like full graph aggregation
(agg_dmpnn_csc) using edge features in an edge-to-edge reduction scheme (e2e)
agg_dmpnn_e2n_fwd( output_edge_embedding: device array, input_edge_embedding: device array, graph: pylibcugraphops.csc_int[32|64], rev_edge_ids: device array, concat_own: bool, stream_id: int = 0 )
- Parameters:
- output_embeddingdevice array type
Device array containing the output node embeddings.
- input_embeddingdevice array type
Device array containing the input node embeddings.
- graphopaque graph type
graph used for the operation.
- rev_edge_idsdevice array
position indices for each edge of the corresponding reverse edge
- concat_ownbool
concat an edge’s input features to the aggregated output features
- stream_idint, default=0
CUDA stream pointer as a python int