PyG or PyTorch Geometric is a deep learning framework for GNNs (graph neural networks). When dealing with graphs, to speed up the computation, we need to do the batch computation, even though the graphs are in different “shapes”. In PyG, it is possible to pack the data in batches. According to the documentation. “Adjacency matricesContinue reading “Deep Learning 20: graph batching in PyTorch Geometric”
