Skip to content

Commit

Permalink
Clarify how to_dense_adj deals with duplicated edges (#8016)
Browse files Browse the repository at this point in the history
Fixes #7998
  • Loading branch information
rusty1s authored Sep 11, 2023
1 parent c2cf68d commit dea1577
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion torch_geometric/utils/to_dense_adj.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,10 @@ def to_dense_adj(
:math:`\mathbf{b} \in {\{ 0, \ldots, B-1\}}^N`, which assigns each
node to a specific example. (default: :obj:`None`)
edge_attr (Tensor, optional): Edge weights or multi-dimensional edge
features. (default: :obj:`None`)
features.
If :obj:`edge_index` contains duplicated edges, the dense adjacency
matrix output holds the summed up entries of :obj:`edge_attr` for
duplicated edges. (default: :obj:`None`)
max_num_nodes (int, optional): The size of the output node dimension.
(default: :obj:`None`)
batch_size (int, optional) The batch size. (default: :obj:`None`)
Expand Down

0 comments on commit dea1577

Please sign in to comment.