About DataParallel , multi gpu #76
Description
In class GraphAttentionLayer, there are two parameter matrices（self.W and self.a）. When I want to use multi GPU, the parameter matrices can only be placed in cuda‘0’，but data of mini batch is located in different cuda. How to solve it?