search

pyg-team / pytorch_geometric

Graph Neural Network Library for PyTorch
https://pyg.org
MIT License
20.94k stars 3.61k forks source link

dropout_adj with SparseTensor? #1604

Open mcd01 opened 4 years ago

mcd01 commented 4 years ago

Hi,

first of all I want to say that I really enjoy working with Pytorch Geometric, it is very powerful and I appreciate the community growing around it. The recently introduced SparseTensor for modeling of adjacency matrices is really nice and helps me to work with far bigger graphs. However, it appears that the dropout_adj is not yet compatible with SparseTensor, thus my question if there is a smart trick to adapt for that (without converting the SparseTensor back to "normal" edge_index format) or at least a good pointer!

Cheers and thanks in advance!

rusty1s commented 4 years ago

I think converting back to edge_index would be fine in this case, but I will think about directly implementing it via SparseTensor.