[Doc] added index [0] to torch.max (dmlc#443)

liushuchun · Mar 8, 2019 · 1327262 · 1327262
1 parent d3cd601
commit 1327262
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/function/reducer.py b/python/dgl/function/reducer.py
@@ -92,6 +92,6 @@ def max(msg, out):
 
     >>> import torch
     >>> def reduce_func(nodes):
-    >>>     return {'h': torch.max(nodes.mailbox['m'], dim=1)}
+    >>>     return {'h': torch.max(nodes.mailbox['m'], dim=1)[0]}
     """
     return SimpleReduceFunction("max", F.max, msg, out)