example.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

#!/usr/bin/env python

from src.scalar import Scalar
from src.graph import Graph

a = Scalar(2, label='a')
b = Scalar(-3, label='b')
c = Scalar(10, label='c')
f = Scalar(-2, label='f')

d = a * b; d.label = 'd'
e = d + c; e.label = 'e'
L = e * f; L.label = 'L'

L.grad = 1.0
e.grad = -2.0
f.grad = 4.0

Graph(L).show()