src/DifferentiableFN.jl

Sat, 20 Feb 2021 16:26:02 -0500

author
Tuomo Valkonen <tuomov@iki.fi>
date
Sat, 20 Feb 2021 16:26:02 -0500
changeset 21
3b7fcc651585
child 25
90f92ee9cb81
permissions
-rw-r--r--

Add DifferentiableFN

21
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
1 module DifferentiableFN
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
2
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
3 using ..LinOps
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
4
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
5 export DiffF,
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
6 value,
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
7 differential,
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
8 adjoint_differential
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
9
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
10 abstract type DiffF{X,Y,T} end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
11
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
12 function value(f :: D, x :: X) :: Y where {X, Y, T <: LinOp{X,Y}, D <: DiffF{X,Y,T}}
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
13 @error "`value` unimplemented"
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
14 end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
15
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
16 # function (f :: D)(x::X) where {X, Y, T <: LinOp{X, Y}, D <: DiffF{X, Y, T}}
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
17 # return value(x)
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
18 # end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
19
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
20 function differential(f :: DiffF{X,Y,T}, x :: X) :: T where {X, Y, T <: LinOp{X, Y}}
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
21 @error "`differential` unimplemented"
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
22 end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
23
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
24 function adjoint_differential(f :: DiffF{X,Y,T}, x :: X) :: T where {X, Y, T <: LinOp{X, Y}}
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
25 @error "`adjoint_differential` unimplemented"
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
26 end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
27
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
28 function adjoint_differential(f :: DiffF{X,Y, T}, x :: X) :: T where {X, Y, T <: AdjointableOp{X, Y}}
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
29 return AdjointOp(differential(f, x))
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
30 end
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
31
3b7fcc651585 Add DifferentiableFN
Tuomo Valkonen <tuomov@iki.fi>
parents:
diff changeset
32 end

mercurial