"torchvision/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "56d4078e53a211dbc18132b798db8e27fdb30ca7"
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
......@@ -197,8 +197,9 @@ class SAGEConv(nn.Module):
Returns
-------
torch.Tensor
The output feature of shape :math:`(N, D_{out})` where :math:`D_{out}`
is size of output feature.
The output feature of shape :math:`(N_{dst}, D_{out})`
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()
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