Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
EdisonLeeeee committed Oct 22, 2023
1 parent 114ddca commit ee041c1
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions torch_geometric/transforms/add_positional_encoding.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,8 +147,6 @@ def forward(self, data: Data) -> Data:
pe_list = [get_self_loop_attr(*to_edge_index(out), num_nodes=N)]
for _ in range(self.walk_length - 1):
out = out @ adj
if out.layout == torch.sparse_coo:
out = out._coalesced_(True)
pe_list.append(get_self_loop_attr(*to_edge_index(out), N))
pe = torch.stack(pe_list, dim=-1)

Expand Down

0 comments on commit ee041c1

Please sign in to comment.