Repository URL to install this package:
Version:
2.4.1 ▾
|
from torch.distributed.device_mesh import ( # noqa: F401
_get_device_handle,
_mesh_resources,
DeviceMesh,
init_device_mesh,
)