Numeric.AD.Internal.Var

Variables used for reverse-mode automatic differentiation.

Documentation

class Primal v => Var v where

Used to mark variables for inspection during the reverse pass

Methods

var :: a -> Int -> v a

varId :: v a -> Int

Instances

Var Variable

Var Kahn

(Primal (AD f), Var f) => Var (AD f)

Primal (Reverse s) => Var (Reverse s)

bind :: (Traversable f, Var v) => f a -> (f (v a), (Int, Int))

unbind :: (Functor f, Var v) => f (v a) -> Array Int a -> f a

unbindMap :: (Functor f, Var v, Num a) => f (v a) -> IntMap a -> f a

unbindWith :: (Functor f, Var v, Num a) => (a -> b -> c) -> f (v a) -> Array Int b -> f c

unbindMapWithDefault :: (Functor f, Var v, Num a) => b -> (a -> b -> c) -> f (v a) -> IntMap b -> f c

data Variable a

Constructors

Variable a !Int

Instances

Primal Variable

Var Variable

vary :: Var f => Variable a -> f a