HoLa-BRep

Runtime error

HoLa-BRep / pointnet2_ops_lib /pointnet2_ops /_ext-src /include /sampling.h

fresh start

e289681 about 1 year ago

266 Bytes

#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);

	#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);