Skip to content

GitLab

Commit 9e048aad authored by rusty1s's avatar rusty1s
Browse files

update doc

parent 34f083a9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
torch_cluster/radius.py
View file @ 9e048aad
...@@ -22,8 +22,7 @@ def radius(x, y, r, batch_x=None, batch_y=None, max_num_neighbors=32): ...@@ -22,8 +22,7 @@ def radius(x, y, r, batch_x=None, batch_y=None, max_num_neighbors=32):
:math:`\mathbf{b} \in {\{ 0, \ldots, B-1\}}^M`, which assigns each :math:`\mathbf{b} \in {\{ 0, \ldots, B-1\}}^M`, which assigns each
node to a specific example. (default: :obj:`None`) node to a specific example. (default: :obj:`None`)
max_num_neighbors (int, optional): The maximum number of neighbors to max_num_neighbors (int, optional): The maximum number of neighbors to
return for each element in :obj:`y`. return for each element in :obj:`y`. (default: :obj:`32`)
This flag is only needed for CUDA tensors. (default: :obj:`32`)
:rtype: :class:`LongTensor` :rtype: :class:`LongTensor`
......
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