torch_harmonics.distributed.azimuth_group_rank#

torch_harmonics.distributed.azimuth_group_rank()[source]#

Return this rank’s index within the azimuth group (0 if not distributed).

Return type:

int