Get Started
Advanced Materials
API Reference
shared_tensor()
Notes
Misc
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