torch_geometric.transforms.GridSampling

class GridSampling(size: Union[float, List[float], Tensor], start: Optional[Union[float, List[float], Tensor]] = None, end: Optional[Union[float, List[float], Tensor]] = None)

Bases: BaseTransform

将点聚类成固定大小的体素 (功能名称: grid_sampling)。 每个返回的聚类是基于给定聚类内所有点的平均值生成的新点。

Parameters:

• size (float or [float] or Tensor) – Size of a voxel (in each dimension).

• start (float 或 [float] 或 Tensor, 可选) – 网格的起始坐标（在每个维度中）。如果设置为 None，将设置为在 data.pos 中找到的最小坐标。 (默认值: None)

• end (float 或 [float] 或 Tensor, 可选) – 网格的结束坐标 （在每个维度中）。如果设置为 None，将设置为 data.pos 中找到的最大坐标。 （默认值：None）