dgl.multiprocessing.pytorch.shared_tensor

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

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

内容未初始化。

参数:

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

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

返回:

共享张量。

返回类型:

张量