dgl.multiprocessing.pytorch.shared_tensor

dgl.multiprocessing.pytorch.shared_tensor(shape, dtype=torch.float32)[source]

创建一个在同一个torch.distributed进程组中所有进程可访问的共享内存张量。

内容未初始化。

Parameters:

  • shape (tuple[int]) – 张量的形状。

  • dtype (torch.dtype, optional) – 张量的数据类型。

Returns:

共享的张量。

Return type:

张量