torch_harmonics.distributed.azimuth_group_size#

torch_harmonics.distributed.azimuth_group_size()[source]#

Return the number of ranks in the azimuth group (1 if not distributed).

Return type:

int