Numeric.AD.Internal.Var

Reverse-Mode Automatic Differentiation using a single tape.

This version uses Data.Reflection to update a single tape.

This is asymptotically faster than using Reverse, which is forced to reify and topologically sort the graph, but it is less friendly to the use of sparks.

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 Reverse

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

Primal (Wengert s) => Var (Wengert 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