torch.cpu

若API“是否支持”“是”“限制与说明”“-”,说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.cpu.current_stream

-

torch.cpu.is_available

-

torch.cpu.synchronize

-

torch.cpu.stream

-

torch.cpu.device_count

-

torch.cpu.StreamContext

-