torch.cpu

API名称

是否支持

限制与说明

torch.cpu.current_stream

是

   

torch.cpu.is_available

是

   

torch.cpu.synchronize

是

   

torch.cpu.stream

是

   

torch.cpu.device_count

是

   

torch.cpu.StreamContext

是

   

torch.cpu.Stream

是

   
父主题： PyTorch原生API