aodhneine/pygrad
1
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

Compare commits

base: aodhneine:a75365a1ab13e22a3e6a465b74139ee023bcef70
Branches Tags
aodhneine:trunk
...
compare: aodhneine:0ad8ac2fcde1ffbc8ef4b418897100fcee1332ed
Branches Tags
aodhneine:trunk

2 commits
a75365a1ab ... 0ad8ac2fcd

Author SHA1 Message Date
Aodhnait Étaín 0ad8ac2fcd Add rule to ignore __pycache__ 2020-11-15 19:05:06 +00:00
Aodhnait Étaín 2755798390 Update README to reflect added support for arrays 2020-11-15 19:04:49 +00:00
2 changed files with 9 additions and 3 deletions
Show stats Expand all files Collapse all files

1
.gitignore vendored
View file

@ -3,3 +3,4 @@ include/
lib/
lib64
pyvenv.cfg
__pycache__/

11
README.md
View file

@ -9,15 +9,20 @@ Neural network library written in Python completely from scratch, using only num
## Usage
```
import nn
from pygrad.tensor import Tensor
a, b = Tensor(2), Tensor(3)
a, b = Tensor(np.random.randn(3, 1)), Tensor(np.random.randn(1, 3))
c = a.mul(b)
d = c.exp()
d.backward()
print(a.grad, b.grad, c.grad)
# should output (1210.2863804782053, 806.8575869854702, 403.4287934927351)
# should output something like
# [[-16.51232262 -16.51232262 -16.51232262]]
# [[-53.88103349 -53.88103349 -53.88103349]]
# [[ 3.50417648 3.50417648 3.50417648]
# [15.67424587 15.67424587 15.67424587]
# [16.0131011 16.0131011 16.0131011 ]]
```
## References