[Bugfix] Fix shape of test_relation tensor in RGCN (#464)

examples/pytorch/rgcn/link_predict.py

@@ -113,7 +113,7 @@ def main(args):
     test_deg = test_graph.in_degrees(
                 range(test_graph.number_of_nodes())).float().view(-1,1)
     test_node_id = torch.arange(0, num_nodes, dtype=torch.long).view(-1, 1)
-    test_rel = torch.from_numpy(test_rel).view(-1, 1)
+    test_rel = torch.from_numpy(test_rel)
     test_norm = torch.from_numpy(test_norm).view(-1, 1)
     test_graph.ndata.update({'id': test_node_id, 'norm': test_norm})
     test_graph.edata['type'] = test_rel