Up – Package index » torch » Library torch » Torch » Device

Module `Torch.Device`Source

torch
- README
- Library torch
  - Torch
- Library torch.core
  - Torch_core
- Library torch.toplevel
  - Torch_toplevel
- Library torch.vision
  - Torch_vision
- Sources

Sourcetype t = Torch_core.Device.t =

| Cpu
| Cuda of int

Sourceval cuda_if_available : unit -> t