Construct a CPU device Parameters ---------- dev_id : int, optional The integer device id Returns ------- dev : Device The created device
(dev_id=0)
| 354 | |
| 355 | |
| 356 | def cpu(dev_id=0): |
| 357 | """Construct a CPU device |
| 358 | |
| 359 | Parameters |
| 360 | ---------- |
| 361 | dev_id : int, optional |
| 362 | The integer device id |
| 363 | |
| 364 | Returns |
| 365 | ------- |
| 366 | dev : Device |
| 367 | The created device |
| 368 | """ |
| 369 | return device(DLDeviceType.kDLCPU, dev_id) |
| 370 | |
| 371 | |
| 372 | def cuda(dev_id=0): |