GroupAll¶
- class mmcv.ops.GroupAll(use_xyz: bool = True)[source]¶
Group xyz with feature.
- Parameters
use_xyz (bool) – Whether to use xyz.
- forward(xyz: torch.Tensor, new_xyz: torch.Tensor, features: Optional[torch.Tensor] = None) → torch.Tensor[source]¶
- Parameters
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.
- Returns
(B, C + 3, 1, N) Grouped feature.
- Return type
Tensor