RaggedToDenseOps.cpp File Reference

(9238339 (Tue Mar 14 18:49:09 2023 -0700))

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

Macros
#define	CALL(value_t, fn)

Functions
torch::Tensor	RaggedToDense (torch::Tensor values, torch::Tensor row_splits, const int64_t out_col_size, torch::Tensor default_value)

Macro Definition Documentation

CALL

#define CALL	(		value_t,
			fn
	)

Value:

    if (CompareTorchDtype<value_t>(value_type)) {                            \
        return fn<value_t>(values, row_splits, out_col_size, default_value); \
    }

Function Documentation

RaggedToDense()

torch::Tensor RaggedToDense	(	torch::Tensor	values,
		torch::Tensor	row_splits,
		const int64_t	out_col_size,
		torch::Tensor	default_value
	)