package torch

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

Module C.OptimizerSource

Sourcetype t = unit Ctypes.ptr
Sourceval adam : (float -> float -> float -> float -> t F.return) F.result
Sourceval rmsprop : (float -> float -> float -> float -> float -> int -> t F.return) F.result
Sourceval sgd : (float -> float -> float -> float -> bool -> t F.return) F.result
Sourceval add_parameters : (t -> Tensor.t Ctypes_static.ptr -> int -> unit F.return) F.result
Sourceval set_learning_rate : (t -> float -> unit F.return) F.result
Sourceval set_momentum : (t -> float -> unit F.return) F.result
Sourceval zero_grad : (t -> unit F.return) F.result
Sourceval step : (t -> unit F.return) F.result
Sourceval free : (t -> unit F.return) F.result