Fix (#2070)

Co-authored-by: Ubuntu <ubuntu@ip-172-31-1-5.us-west-2.compute.internal>

examples/pytorch/transformer/modules/models.py

@@ -79,9 +79,8 @@ class Transformer(nn.Module):
         g.apply_edges(src_dot_dst('k', 'q', 'score'), eids)
         g.apply_edges(scaled_exp('score', np.sqrt(self.d_k)), eids)
         # Send weighted values to target nodes
-        g.send_and_recv(eids,
-                        [fn.src_mul_edge('v', 'score', 'v'), fn.copy_edge('score', 'score')],
-                        [fn.sum('v', 'wv'), fn.sum('score', 'z')])
+        g.send_and_recv(eids, fn.src_mul_edge('v', 'score', 'v'), fn.sum('v', 'wv'))
+        g.send_and_recv(eids, fn.copy_edge('score', 'score'), fn.sum('score', 'z'))
 
     def update_graph(self, g, eids, pre_pairs, post_pairs):
         "Update the node states and edge states of the graph."