#include <vector>
#include "open3d/ml/pytorch/TorchHelper.h"
#include "open3d/ml/pytorch/pointnet/BallQueryKernel.h"
#include "torch/script.h"