- class CuGraphSAGEConv(in_channels: int, out_channels: int, aggr: str = 'mean', normalize: bool = False, root_weight: bool = True, project: bool = False, bias: bool = True)
The GraphSAGE operator from the “Inductive Representation Learning on Large Graphs” paper.
- forward(x: Tensor, csc: Tuple[Tensor, Tensor, int], max_num_neighbors: Optional[int] = None) Tensor
Runs the forward pass of the module.
x (torch.Tensor) – The node features.
csc ((torch.Tensor, torch.Tensor, int)) – A tuple containing the CSC representation of a graph, given as a tuple of
(row, colptr, num_src_nodes). Use the
CuGraphModule.to_csc()method to convert an
edge_indexrepresentation to the desired format.
max_num_neighbors (int, optional) – The maximum number of neighbors of a target node. It is only effective when operating in a bipartite graph. When not given, the value will be computed on-the-fly, leading to slightly worse performance. (default:
Resets all learnable parameters of the module.