package torch

  1. Overview
  2. Docs

Module C.OptimizerSource

Sourcetype t = unit Ctypes.ptr
Sourceval adam : (float -> float -> float -> float -> t Torch_generated.return) Torch_generated.result
Sourceval rmsprop : (float -> float -> float -> float -> float -> int -> t Torch_generated.return) Torch_generated.result
Sourceval sgd : (float -> float -> float -> float -> bool -> t Torch_generated.return) Torch_generated.result
Sourceval set_learning_rate : (t -> float -> unit Torch_generated.return) Torch_generated.result
Sourceval set_momentum : (t -> float -> unit Torch_generated.return) Torch_generated.result