dgl.multiprocessing.pytorch.call_once_and_share

dgl.multiprocessing.pytorch.call_once_and_share(func, shape, dtype, rank=0)

在PyTorch分布式进程组的单个进程中调用函数，并与其他进程共享结果。

Parameters:

• func (可调用) – 任何不接受参数并返回任意对象的可调用对象。

• shape (tuple[int]) – 共享张量的形状。必须与 func 的输出匹配。

• dtype (torch.dtype) – 共享张量的数据类型。必须与 func 的输出匹配。

• rank (int, optional) – 实际执行函数的进程ID。