GroupAll¶
- class mmcv.ops.GroupAll(use_xyz: bool = True)[source]¶
Group xyz with feature.
- Parameters:
use_xyz (bool) – Whether to use xyz.
- forward(xyz: Tensor, new_xyz: Tensor, features: Tensor | None = None) 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