InvertNeighborsListOpKernel.h File Reference

(b9e049c (Sun Dec 31 11:36:26 2023 -0800))

#include "torch/script.h"

Go to the source code of this file.

Functions
template<class TIndex , class TAttr >
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	InvertNeighborsListCPU (int64_t num_points, const torch::Tensor &inp_neighbors_index, const torch::Tensor &inp_neighbors_row_splits, const torch::Tensor &inp_neighbors_attributes)

Function Documentation

InvertNeighborsListCPU()

template<class TIndex , class TAttr >

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> InvertNeighborsListCPU	(	int64_t	num_points,
		const torch::Tensor &	inp_neighbors_index,
		const torch::Tensor &	inp_neighbors_row_splits,
		const torch::Tensor &	inp_neighbors_attributes
	)