MCPcopy Create free account
hub / github.com/apple/ml-pointersect / keep_min_k_values

Function keep_min_k_values

pointersect/pr/cuda/pr_cuda.cpp:550–559  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

548
549
550torch::Tensor keep_min_k_values(
551 torch::Tensor values, // (b, n), float
552 int64_t k
553) {
554// return values;
555 return keep_min_k_values_cuda(
556 values, // (b, n,
557 k
558 );
559}
560
561
562// Find the k nearest points within `radius` of a ray, ie, the vertical distance from the point to ray <= radius.

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected