You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+1-1Lines changed: 1 addition & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -164,7 +164,7 @@ Computes graph edges to the nearest *k* points.
164
164
**Args:**
165
165
166
166
***x***(Tensor)*: Node feature matrix of shape `[N, F]`.
167
-
***r***(float)*: The radius.
167
+
***k***(int)*: The number of neighbors.
168
168
***batch***(LongTensor, optional)*: Batch vector of shape `[N]`, which assigns each node to a specific example. `batch` needs to be sorted. (default: `None`)
169
169
***loop***(bool, optional)*: If `True`, the graph will contain self-loops. (default: `False`)
170
170
***flow***(string, optional)*: The flow direction when using in combination with message passing (`"source_to_target"` or `"target_to_source"`). (default: `"source_to_target"`)
0 commit comments