Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "72780ff5b154f37194903078ef6caa5d65c653e3"
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): ...@@ -92,6 +92,6 @@ def max(msg, out):
>>> import torch >>> import torch
>>> def reduce_func(nodes): >>> 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) 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