leffff/graphormer-pyg

Optimize for loop in EdgeEncoding

leffff opened this issue · 0 comments

leffff commented

graphormer-pyg/graphormer/layers.py

Line 70 in 951cad0

def forward(self, x: torch.Tensor, edge_attr: torch.Tensor, edge_paths) -> torch.Tensor: