pytorch3d.ops.ball_query 用法

官网介绍：

输入是两个点云P1，P2（也可以是同一个，比如都是P1），

举个例子：

import torch
from pytorch3d.ops import ball_query

# 示例点云数据
N = 2  # 批次大小
P1 = 5  # 源点云中的点数
P2 = 3  # 查询点云中的点数
p1 = torch.rand(N, P1