pytorchGraph Neural Network Pool

pytorch - Graph Neural Network Pooling for keeping the same shape of node without batching - Stack Overflow
I want to add a pooling layer on my model. I want to preserve the num of node as I m doing node predict
pytorchGraph Neural Network Pooling for keeping the same shape of node without batchingStack Overflow
admin14天前
30