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