here got the the error info
rst = rst + resval RuntimeError: The size of tensor a (128) must match the size of tensor b (1757) at non-singleton dimension 0
so I think this line should change to
resval = self.res_fc(h_dst).view(h_dst.shape, -1, self._out_feats)[:graph.number_of_dst_nodes()]
the original code line is
Am I right?