torch.distributed.tensor.parallel

API名称

是否支持

限制与说明

torch.distributed.tensor.parallel.fsdp.enable_2d_with_fsdp

否

   
父主题： PyTorch原生API