| Copyright | (c) Sam Stites 2017 |
|---|---|
| License | BSD3 |
| Maintainer | sam@stites.io |
| Stability | experimental |
| Portability | non-portable |
| Safe Haskell | None |
| Language | Haskell2010 |
Torch.Indef.Static.NN.Activation
Description
Synopsis
- _pReLU_updateOutput :: Tensor d -> Tensor d -> Tensor d -> IO ()
- _pReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> IO ()
- _pReLU_accGradParameters :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> Tensor d -> Double -> IO ()
- _rReLU_updateOutput :: Tensor d -> Tensor d -> Tensor d -> Double -> Double -> Bool -> Bool -> Generator -> IO ()
- _rReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> Double -> Double -> Bool -> Bool -> IO ()
- _eLU_updateOutput :: Tensor d -> Tensor d -> Double -> Double -> Bool -> IO ()
- _eLU_updateGradInput :: Tensor d -> Tensor d' -> Tensor d'' -> Double -> Double -> IO ()
- _leakyReLU_updateOutput :: Tensor d -> Tensor d -> Double -> Bool -> IO ()
- _leakyReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Double -> Bool -> IO ()
- relu :: Reifies s W => Dimensions d => BVar s (Tensor d) -> BVar s (Tensor d)
- reluIO :: Dimensions d => Tensor d -> IO (Tensor d, Tensor d -> IO (Tensor d))
- threshold :: forall s d. Reifies s W => Dimensions d => Double -> Double -> BVar s (Tensor d) -> BVar s (Tensor d)
- thresholdIO :: forall d. Dimensions d => Double -> Double -> Tensor d -> IO (Tensor d, Tensor d -> IO (Tensor d))
Documentation
_pReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> IO () Source #
pReLU updateGradInput
_pReLU_accGradParameters :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> Tensor d -> Double -> IO () Source #
pReLU accGradParameters
_rReLU_updateOutput :: Tensor d -> Tensor d -> Tensor d -> Double -> Double -> Bool -> Bool -> Generator -> IO () Source #
rReLU updateOutput
_rReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Tensor d -> Double -> Double -> Bool -> Bool -> IO () Source #
rReLU updateGradInput
_eLU_updateOutput :: Tensor d -> Tensor d -> Double -> Double -> Bool -> IO () Source #
eLU updateOutput
_eLU_updateGradInput :: Tensor d -> Tensor d' -> Tensor d'' -> Double -> Double -> IO () Source #
eLU updateGradInput
_leakyReLU_updateOutput :: Tensor d -> Tensor d -> Double -> Bool -> IO () Source #
leakyReLU updateOutput
_leakyReLU_updateGradInput :: Tensor d -> Tensor d -> Tensor d -> Double -> Bool -> IO () Source #
leakyReLU updateGradInput
relu :: Reifies s W => Dimensions d => BVar s (Tensor d) -> BVar s (Tensor d) Source #
ReLU activation function
reluIO :: Dimensions d => Tensor d -> IO (Tensor d, Tensor d -> IO (Tensor d)) Source #
ReLU activation function