Documentation

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

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

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

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

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

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

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

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

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

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

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

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