Skip to content

GitLab

Unverified Commit 0851b4df authored by WangYQ's avatar WangYQ Committed by GitHub
Browse files

Sageconv doc (#3093) 

* revise the doc of forward function in SAGEConv

* remove trailing space
parent 9c41c22d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
python/dgl/nn/pytorch/conv/sageconv.py
View file @ 0851b4df
...@@ -197,8 +197,9 @@ class SAGEConv(nn.Module): ...@@ -197,8 +197,9 @@ class SAGEConv(nn.Module):
Returns Returns
------- -------
torch.Tensor torch.Tensor
The output feature of shape :math:`(N, D_{out})` where :math:`D_{out}` The output feature of shape :math:`(N_{dst}, D_{out})`
is size of output feature. where :math:`N_{dst}` is the number of destination nodes in the input graph,
math:`D_{out}` is size of output feature.
""" """
self._compatibility_check() self._compatibility_check()
with graph.local_scope(): with graph.local_scope():
......
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