Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "989702b3638c4cb4b0690f906f688cde28821bc0"
Commit 13272628 authored by mc-robinson's avatar mc-robinson Committed by Minjie Wang
Browse files

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

parent d3cd6014
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/function/reducer.py
View file @ 13272628
......@@ -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)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment