InterpolateOps.cpp File Reference

(d32b4fc (Mon Sep 14 15:46:25 2026 -0700))

#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <tuple>
#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/pointnet/InterpolateKernel.h"
#include "torch/script.h"

Functions
std::tuple< torch::Tensor, torch::Tensor >	three_nn (torch::Tensor query_pts, torch::Tensor data_pts)
torch::Tensor	three_interpolate (torch::Tensor points, torch::Tensor idx, torch::Tensor weights)
torch::Tensor	three_interpolate_grad (torch::Tensor grad_out, torch::Tensor idx, torch::Tensor weights, const int64_t M)

Function Documentation

three_interpolate()

torch::Tensor three_interpolate	(	torch::Tensor	points,
		torch::Tensor	idx,
		torch::Tensor	weights
	)

three_interpolate_grad()

torch::Tensor three_interpolate_grad	(	torch::Tensor	grad_out,
		torch::Tensor	idx,
		torch::Tensor	weights,
		const int64_t	M
	)

three_nn()

std::tuple< torch::Tensor, torch::Tensor > three_nn	(	torch::Tensor	query_pts,
		torch::Tensor	data_pts
	)