torch_harmonics.distributed.reduce_from_azimuth_region#
- torch_harmonics.distributed.reduce_from_azimuth_region(input_)[source]#
All-reduce across azimuth ranks in the forward pass; identity in the backward pass.
Azimuth counterpart of
reduce_from_polar_region().- Parameters:
input (torch.Tensor) – Partial result on the local azimuth rank.
- Returns:
Sum of
input_across all azimuth ranks.- Return type: