torch.distributed.tensor.parallel API名称 是否支持 限制与说明 torch.distributed.tensor.parallel.fsdp.enable_2d_with_fsdp 否 父主题: PyTorch原生API