Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "e489abc684c864f1da010d56e3ca66dbd2df82fb"
Unverified Commit 5bc029b9 authored by Zihao Ye's avatar Zihao Ye Committed by GitHub
Browse files

upd (#2097)

parent 163398c0
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 1 deletion
docs/source/api/python/dgl.ops.rst
View file @ 5bc029b9
...@@ -15,7 +15,7 @@ It *fuses* two steps into one kernel. ...@@ -15,7 +15,7 @@ It *fuses* two steps into one kernel.
1. Computes messages by add/sub/mul/div source node and edge features, 1. Computes messages by add/sub/mul/div source node and edge features,
or copy node features to edges. or copy node features to edges.
2. Aggregate the messages by sum/max/min as the features on destination nodes. 2. Aggregate the messages by sum/max/min/mean as the features on destination nodes.
Our implementation supports tensors on CPU/GPU in PyTorch/MXNet/Tensorflow Our implementation supports tensors on CPU/GPU in PyTorch/MXNet/Tensorflow
as input. All operators are equipped with autograd (computing the input gradients as input. All operators are equipped with autograd (computing the input gradients
...@@ -100,12 +100,18 @@ graph. ...@@ -100,12 +100,18 @@ graph.
u_sub_e_min u_sub_e_min
u_mul_e_min u_mul_e_min
u_div_e_min u_div_e_min
u_add_e_mean
u_sub_e_mean
u_mul_e_mean
u_div_e_mean
copy_u_sum copy_u_sum
copy_e_sum copy_e_sum
copy_u_max copy_u_max
copy_e_max copy_e_max
copy_u_min copy_u_min
copy_e_min copy_e_min
copy_u_mean
copy_e_mean
GSDDMM functions GSDDMM functions
---------------- ----------------
......
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