package torch
Legend:
Library
Module
Module type
Parameter
Class
Class type
type t = unit Ctypes.ptr
val t : t Ctypes.typ
val adam : (float -> float -> float -> float -> float -> t F.return) F.result
val rmsprop : (float -> float -> float -> float -> float -> int -> t F.return) F.result
val sgd : (float -> float -> float -> float -> bool -> t F.return) F.result
val add_parameters : (t -> Tensor.t Ctypes_static.ptr -> int -> unit F.return) F.result
val set_learning_rate : (t -> float -> unit F.return) F.result
val set_momentum : (t -> float -> unit F.return) F.result
val zero_grad : (t -> unit F.return) F.result
val step : (t -> unit F.return) F.result
val free : (t -> unit F.return) F.result