torch_harmonics.distributed.reduce_from_polar_region

torch_harmonics.distributed.reduce_from_polar_region(input_)

All-reduce across polar ranks in the forward pass; identity in the backward pass.

Use this to aggregate partial results computed independently on each polar rank.

Parameters:

input (torch.Tensor) – Partial result on the local polar rank.

Returns:

Sum of input_ across all polar ranks.

Return type:

torch.Tensor