sdf_torch.h

浏览该文件的文档.

#pragma once
#include "vde/sdf/sdf_gradient.h"
#include "vde/sdf/sdf_tree.h"
#include "vde/core/point.h"
#include <vector>
 
// This header is optional — only compiled if Torch is available
 
namespace vde::sdf {
 
void evaluate_batch(const SdfNodePtr& root,
                    const double* points, int n,
                    double* distances);
 
void gradient_batch(const SdfNodePtr& root,
                    const double* points, int n,
                    double* gradients);
 
void evaluate_with_gradient_batch(const SdfNodePtr& root,
                                   const double* points, int n,
                                   double* distances,
                                   double* gradients);
 
} // namespace vde::sdf