GroupAll¶
- class mmcv.ops.GroupAll(use_xyz: bool = True)[源代码]¶
Group xyz with feature.
- 参数
use_xyz (bool) – Whether to use xyz.
- forward(xyz: torch.Tensor, new_xyz: torch.Tensor, features: Optional[torch.Tensor] = None) → torch.Tensor[源代码]¶
- 参数
xyz (Tensor) – (B, N, 3) xyz coordinates of the features.
new_xyz (Tensor) – new xyz coordinates of the features.
features (Tensor) – (B, C, N) features to group.
- 返回
(B, C + 3, 1, N) Grouped feature.
- 返回类型
Tensor