VoxelizeOps.cpp File Reference

(313315d (Wed Dec 23 23:39:47 2020 -0800))

#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/misc/VoxelizeOpKernel.h"
#include "torch/script.h"

Macros
#define	CALL(point_t, fn)

Functions
std::tuple< torch::Tensor, torch::Tensor, torch::Tensor >	Voxelize (torch::Tensor points, torch::Tensor voxel_size, torch::Tensor points_range_min, torch::Tensor points_range_max, const int64_t max_points_per_voxel, const int64_t max_voxels)

Macro Definition Documentation

CALL

#define CALL	(		point_t,
			fn
	)

Value:

if (CompareTorchDtype<point_t>(points_dtype)) {                         \
        fn<point_t>(points, voxel_size, points_range_min, points_range_max, \
                    max_points_per_voxel, max_voxels, voxel_coords,         \
                    voxel_point_indices, voxel_point_row_splits);           \
        return std::make_tuple(voxel_coords, voxel_point_indices,           \
                               voxel_point_row_splits);                     \
    }

Function Documentation

Voxelize()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor> Voxelize	(	torch::Tensor	points,
		torch::Tensor	voxel_size,
		torch::Tensor	points_range_min,
		torch::Tensor	points_range_max,
		const int64_t	max_points_per_voxel,
		const int64_t	max_voxels
	)