7 lines
260 B
C
7 lines
260 B
C
#pragma once
|
|
#include <torch/extension.h>
|
|
|
|
at::Tensor gather_points(at::Tensor points, at::Tensor idx);
|
|
at::Tensor gather_points_grad(at::Tensor grad_out, at::Tensor idx, const int n);
|
|
at::Tensor furthest_point_sampling(at::Tensor points, const int nsamples);
|