Gograd is a small automatic reverse-mode differentiation framework written in Go.
a := NewVar(4)
b := NewVar(3)
c := a.Mul(a.Add(b))
c.Backward()
fmt.Println(c.value)
// prints 28
fmt.Println(a.grad)
// prints 11
fmt.Println(b.grad)
// prints 4-
Notifications
You must be signed in to change notification settings - Fork 0
Gograd is a small automatic differentiation framework written in Go.
License
lucaionescu/gograd
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Gograd is a small automatic differentiation framework written in Go.