Skip to content

[FEA] Option to have indices/distances arrays returned on cpu memory in all_neighbors #1833

Description

@jinsolp

Current version of all_neighbors always returns indices/distances on GPU when it doesn't have to.
Allow user to choose whether they want returned indices/distances on GPU VS CPU.

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions