VoxelizeOps.cpp File Reference

(252c867 (Sun May 5 05:39:34 2024 +0000))

#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, torch::Tensor >	Voxelize (torch::Tensor points, torch::Tensor row_splits, 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, row_splits, voxel_size, points_range_min,          \
                    points_range_max, max_points_per_voxel, max_voxels,        \
                    voxel_coords, voxel_point_indices, voxel_point_row_splits, \
                    voxel_batch_splits);                                       \
        return std::make_tuple(voxel_coords, voxel_point_indices,              \
                               voxel_point_row_splits, voxel_batch_splits);    \
    }

Function Documentation

Voxelize()

std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> Voxelize	(	torch::Tensor	points,
		torch::Tensor	row_splits,
		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
	)