module Numeric.AD.Mode.Directed
(
grad
, grad'
, jacobian
, jacobian'
, diff
, diff'
, Direction(..)
) where
import Prelude hiding (reverse)
import Numeric.AD.Mode
import Data.Traversable (Traversable)
import qualified Numeric.AD.Mode.Kahn as K
import qualified Numeric.AD.Mode.Forward as F
import qualified Numeric.AD.Mode.Tower as T
import qualified Numeric.AD.Mode.Reverse as R
import qualified Numeric.AD as M
import Data.Ix
data Direction
= Forward
| Kahn
| Reverse
| Tower
| Mixed
deriving (Show, Eq, Ord, Read, Bounded, Enum, Ix)
diff :: Num a => Direction -> (forall t. Mode t => t -> t) -> a -> a
diff Forward f a = F.diff f a
diff Kahn f a = K.diff f a
diff Reverse f a = R.diff f a
diff Tower f a = T.diff f a
diff Mixed f a = F.diff f a
diff' :: Num a => Direction -> (forall t. Mode t => t -> t) -> a -> (a, a)
diff' Forward f a = F.diff' f a
diff' Kahn f a = K.diff' f a
diff' Reverse f a = R.diff' f a
diff' Tower f a = T.diff' f a
diff' Mixed f a = F.diff' f a
jacobian :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f (t) -> g (t)) -> f a -> g (f a)
jacobian Forward f a = F.jacobian f a
jacobian Kahn f a = K.jacobian f a
jacobian Reverse f a = R.jacobian f a
jacobian Tower f a = F.jacobian f a
jacobian Mixed f a = M.jacobian f a
jacobian' :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f (t) -> g (t)) -> f a -> g (a, f a)
jacobian' Forward f a = F.jacobian' f a
jacobian' Kahn f a = K.jacobian' f a
jacobian' Reverse f a = R.jacobian' f a
jacobian' Tower f a = F.jacobian' f a
jacobian' Mixed f a = M.jacobian' f a
grad :: (Traversable f, Num a) => Direction -> (forall t. Mode t => f (t) -> t) -> f a -> f a
grad Forward f a = F.grad f a
grad Kahn f a = K.grad f a
grad Reverse f a = R.grad f a
grad Tower f a = F.grad f a
grad Mixed f a = M.grad f a
grad' :: (Traversable f, Num a) => Direction -> (forall t. Mode t => f (t) -> t) -> f a -> (a, f a)
grad' Forward f a = F.grad' f a
grad' Kahn f a = K.grad' f a
grad' Reverse f a = R.grad' f a
grad' Tower f a = F.grad' f a
grad' Mixed f a = M.grad' f a