Documentation

adagrad :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

rmsprop :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

sgd :: CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

adam :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

adamw :: CDouble -> CDouble -> CDouble -> CDouble -> CDouble -> CBool -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

lbfgs :: CDouble -> CInt -> CInt -> CDouble -> CDouble -> CInt -> Maybe (ForeignPtr StdString) -> ForeignPtr TensorList -> IO (ForeignPtr Optimizer) Source #

getParams :: ForeignPtr Optimizer -> IO (ForeignPtr TensorList) Source #

step :: ForeignPtr Optimizer -> (ForeignPtr TensorList -> IO (ForeignPtr Tensor)) -> IO (ForeignPtr Tensor) Source #

stepWithGenerator :: ForeignPtr Optimizer -> ForeignPtr Generator -> (ForeignPtr TensorList -> ForeignPtr Generator -> IO (ForeignPtr (StdTuple '(Tensor, Generator)))) -> IO (ForeignPtr (StdTuple '(Tensor, Generator))) Source #

unsafeStep :: ForeignPtr Optimizer -> ForeignPtr Tensor -> IO (ForeignPtr TensorList) Source #

save :: ForeignPtr Optimizer -> ForeignPtr StdString -> IO () Source #

load :: ForeignPtr Optimizer -> ForeignPtr StdString -> IO () Source #