package torch

  1. Overview
  2. Docs
package torch
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
include Sexplib0.Sexpable.S with type t := t
Sourceval t_of_sexp : Sexplib0.Sexp.t -> t
Sourceval sexp_of_t : t -> Sexplib0.Sexp.t
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.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct