1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
// RUN: %target-run-simple-swift
// REQUIRES: executable_test
import _Differentiation
public func simWithPullback(params: MinParams) -> (value: Output, pullback: (Output.TangentVector) -> (MinParams.TangentVector)){
let simulationValueAndPullback = valueWithPullback(at: params, of: run)
return (value: simulationValueAndPullback.value, pullback: simulationValueAndPullback.pullback)
}
@differentiable(reverse)
public func run(params: MinParams) -> Output {
for t in 0 ... 1 {
}
let res = MiniLoop(other: params._twoDArray).results
return Output(results: res)
}
struct MiniLoop: Differentiable {
var results: Float
var twoDArray: Float
@differentiable(reverse)
init(results: Float = 146, other: Float = 123) {self.results = results; self.twoDArray = other}
}
public struct Output: Differentiable {
public var results: Float
@differentiable(reverse)
public init(results: Float) {self.results = results}
}
public struct MinParams: Differentiable {
public var _twoDArray: Float
public init(foo: Float = 42) { _twoDArray = foo }
}
func test() {
let valueAndPullback = simWithPullback(params: MinParams())
let output = valueAndPullback.value
let resultOnes = Float(1.0)
var grad = valueAndPullback.pullback(Output.TangentVector(results: resultOnes))
print(grad)
grad = valueAndPullback.pullback(Output.TangentVector(results: resultOnes))
print(grad)
grad = valueAndPullback.pullback(Output.TangentVector(results: resultOnes))
print(grad)
}
test()
|
