Open3D (C++ API)
0.12.0
|
#include <vector>
#include "open3d/ml/impl/misc/VoxelPooling.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"
Data Structures | |
class | VoxelPoolingFunction |
Macros | |
#define | FN_PARAMETERS positions, features, voxel_size, position_fn, feature_fn, debug |
#define | CALL(real_t, feat_t, fn) |
#define | FN_PARAMETERS |
#define | CALL(real_t, feat_t, fn) |
Functions | |
template<class TReal , class TFeat > | |
std::tuple< torch::Tensor, torch::Tensor > | VoxelPoolingCPU (const torch::Tensor &positions, const torch::Tensor &features, const double voxel_size, const AccumulationFn position_fn, const AccumulationFn feature_fn, const bool debug) |
template<class TReal , class TFeat > | |
void | VoxelPoolingGradCPU (torch::Tensor &features_backprop, const torch::Tensor &positions, const torch::Tensor &features, const torch::Tensor &pooled_positions, const torch::Tensor &pooled_features_gradient, const double voxel_size, const AccumulationFn position_fn, const AccumulationFn feature_fn) |
std::tuple< torch::Tensor, torch::Tensor > | VoxelPooling (const torch::Tensor &positions, const torch::Tensor &features, const double voxel_size, const std::string &position_fn_str, const std::string &feature_fn_str, const bool debug) |
#define CALL | ( | real_t, | |
feat_t, | |||
fn | |||
) |
#define CALL | ( | real_t, | |
feat_t, | |||
fn | |||
) |
#define FN_PARAMETERS positions, features, voxel_size, position_fn, feature_fn, debug |
#define FN_PARAMETERS |
std::tuple<torch::Tensor, torch::Tensor> VoxelPooling | ( | const torch::Tensor & | positions, |
const torch::Tensor & | features, | ||
const double | voxel_size, | ||
const std::string & | position_fn_str, | ||
const std::string & | feature_fn_str, | ||
const bool | debug | ||
) |
std::tuple<torch::Tensor, torch::Tensor> VoxelPoolingCPU | ( | const torch::Tensor & | positions, |
const torch::Tensor & | features, | ||
const double | voxel_size, | ||
const AccumulationFn | position_fn, | ||
const AccumulationFn | feature_fn, | ||
const bool | debug | ||
) |
void VoxelPoolingGradCPU | ( | torch::Tensor & | features_backprop, |
const torch::Tensor & | positions, | ||
const torch::Tensor & | features, | ||
const torch::Tensor & | pooled_positions, | ||
const torch::Tensor & | pooled_features_gradient, | ||
const double | voxel_size, | ||
const AccumulationFn | position_fn, | ||
const AccumulationFn | feature_fn | ||
) |