torch_harmonics.distributed.copy_to_azimuth_region#
- torch_harmonics.distributed.copy_to_azimuth_region(input_)[source]#
Identity in the forward pass; all-reduce across azimuth ranks in the backward pass.
Azimuth counterpart of
copy_to_polar_region().- Parameters:
input (torch.Tensor) – Replicated tensor (same value on every azimuth rank).
- Returns:
Same tensor (forward is a no-op).
- Return type: