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 |
是 |
|
torch.cpu.Stream |
是 |
父主题: PyTorch原生API支持度