package torch

  1. Overview
  2. Docs
package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type
type t
val adam : learning_rate:Base.float -> beta1:Base.float -> beta2:Base.float -> weight_decay:Base.float -> eps:Base.float -> t
val rmsprop : learning_rate:Base.float -> alpha:Base.float -> eps:Base.float -> weight_decay:Base.float -> momentum:Base.float -> centered:Base.bool -> t
val sgd : learning_rate:Base.float -> momentum:Base.float -> dampening:Base.float -> weight_decay:Base.float -> nesterov:Base.bool -> t
val set_learning_rate : t -> Base.float -> Base.unit
val set_momentum : t -> Base.float -> Base.unit
val add_parameters : t -> Tensor.t Base.list -> Base.unit
val zero_grad : t -> Base.unit
val step : t -> Base.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