Copyright | (c) Sam Stites 2017 |
---|---|
License | BSD3 |
Maintainer | sam@stites.io |
Stability | experimental |
Portability | non-portable |
Safe Haskell | None |
Language | Haskell2010 |
DYNAMIC-NN MODULE WARNING: this module is mostly unfinished and undocumented. It provides, in essence, direct calls to the torch neural network libraries: THNN and THCUNN. Because the dynamic tensor code requires a lot of runtime checks which requires a lot of thought regarding a good API, the recommended route is to use Static tensors, which have a much more natural API and is inherently safer.
Synopsis
- _threshold_updateOutput :: Dynamic -> Dynamic -> Double -> Double -> Bool -> IO ()
- _threshold_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> IO ()
- _pReLU_updateOutput :: Dynamic -> Dynamic -> Dynamic -> IO ()
- _pReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> IO ()
- _pReLU_accGradParameters :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> Dynamic -> Double -> IO ()
- _rReLU_updateOutput :: Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> Bool -> Generator -> IO ()
- _rReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> Bool -> IO ()
- _leakyReLU_updateOutput :: Dynamic -> Dynamic -> Double -> Bool -> IO ()
- _leakyReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Double -> Bool -> IO ()
- _eLU_updateOutput :: Dynamic -> Dynamic -> Double -> Double -> Bool -> IO ()
- _eLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Double -> Double -> IO ()
Documentation
_threshold_updateOutput :: Dynamic -> Dynamic -> Double -> Double -> Bool -> IO () Source #
threshold forward pass (updates the output tensor)
_threshold_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> IO () Source #
threshold backward-update (updates the layer and bias tensors)
_pReLU_updateOutput :: Dynamic -> Dynamic -> Dynamic -> IO () Source #
pReLU forward pass (updates the output tensor)
_pReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> IO () Source #
pReLU backward-update (updates the layer and bias tensors)
_pReLU_accGradParameters :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> Dynamic -> Double -> IO () Source #
pReLU backward-update (updates the layer and bias tensors). Called accGradParameters
in C to indicate accumulating the gradient parameters.
_rReLU_updateOutput :: Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> Bool -> Generator -> IO () Source #
rReLU forward pass (updates the output tensor)
_rReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Dynamic -> Double -> Double -> Bool -> Bool -> IO () Source #
rReLU backward-update (updates the layer and bias tensors)
_leakyReLU_updateOutput :: Dynamic -> Dynamic -> Double -> Bool -> IO () Source #
leakyReLU forward pass (updates the output tensor)
_leakyReLU_updateGradInput :: Dynamic -> Dynamic -> Dynamic -> Double -> Bool -> IO () Source #
leakyReLU backward-update (updates the layer and bias tensors)