當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python PyTorch invoke_on_rank_and_broadcast_result用法及代碼示例


本文簡要介紹python語言中 torchrec.distributed.collective_utils.invoke_on_rank_and_broadcast_result 的用法。

用法:

torchrec.distributed.collective_utils.invoke_on_rank_and_broadcast_result(pg: torch._C._distributed_c10d.ProcessGroup, rank: int, func: Callable[[...], torchrec.distributed.collective_utils.T], *args: Any, **kwargs: Any) → torchrec.distributed.collective_utils.T

調用指定等級的函數並將結果廣播給組內的所有成員。

例子：

id = invoke_on_rank_and_broadcast_result(pg, 0, allocate_id)

相關用法


注：本文由純淨天空篩選整理自pytorch.org大神的英文原創作品 torchrec.distributed.collective_utils.invoke_on_rank_and_broadcast_result。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。