Open3D (C++ API)
0.12.0
|
#include "open3d/ml/impl/misc/VoxelPooling.h"
#include "open3d/ml/pytorch/TorchHelper.h"
#include "torch/script.h"
Macros | |
#define | INSTANTIATE(TReal, TFeat) |
#define | INSTANTIATE(TReal, TFeat) |
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) |
#define INSTANTIATE | ( | TReal, | |
TFeat | |||
) |
#define INSTANTIATE | ( | TReal, | |
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 | ||
) |
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 | ||
) |