Skip to content

Commit

Permalink
Update lab.md
Browse files Browse the repository at this point in the history
  • Loading branch information
TheMrGhostman committed Nov 22, 2024
1 parent b114b67 commit 57cc0ca
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions docs/src/lecture_09/lab.md
Original file line number Diff line number Diff line change
Expand Up @@ -60,12 +60,12 @@ The types of tangents and cotangents depend on the types of the primals. However


!!! warning "Exercise"
```@example labO9
```@example lab09
using ChainRulesCore, ChainRules, ChainRulesTestUtils
```
Write custom `rrule` for following function $f(x,y) = x^2 + 3y$.

```@example labO9
```@example lab09
f1(x::T, y::T) where T<: Real = x^2 + 3*y
```
You can test your solution using
Expand Down Expand Up @@ -157,7 +157,7 @@ The functions `pool_naive` and `pool_sum` perform the same operation with the sa
function pool_sum_pullback(ȳ)
...
end
return y, pool_custom_pullback
return y, pool_sum_pullback
end

```
Expand Down

0 comments on commit 57cc0ca

Please sign in to comment.