package torch

  1. Overview
  2. Docs
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Torch.DeviceSource

Sourcetype t = Torch_core.Device.t =
  1. | Cpu
  2. | Cuda of int
Sourceval cuda_if_available : unit -> t
Sourceval is_cuda : t -> bool
Sourceval get_num_threads : unit -> int
Sourceval set_num_threads : int -> unit
OCaml

Innovation. Community. Security.