A bug of GAT to confirm

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[0], -1, self._out_feats)[:graph.number_of_dst_nodes()]

the original code line is

Am I right?

Can you share a code snippet for reproducing the error?