Get Started
Advanced Materials
API Reference
Notes
Misc
dgl.multiprocessing.pytorch.
shared_tensor
Create a tensor in shared memory accessible by all processes within the same torch.distributed process group.
torch.distributed
The content is uninitialized.
shape (tuple[int]) – The shape of the tensor.
dtype (torch.dtype, optional) – The dtype of the tensor.
The shared tensor.
Tensor