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

Compare commits

base: aodhneine:f85b9670b2bcf48b6f0bc6c8906f7d8ee97964b6
Branches Tags
aodhneine:trunk
...
compare: aodhneine:52d266ef91f7a075f5e0dfae94d28753ef659ca4
Branches Tags
aodhneine:trunk

2 commits
f85b9670b2 ... 52d266ef91

Author SHA1 Message Date
Aodhnait Étaín 52d266ef91 Fix tensor.add to diff. matrices properly 2020-11-15 19:45:51 +00:00
Aodhnait Étaín c1c48bb693 Update nn.test 2020-11-15 19:45:05 +00:00
2 changed files with 2 additions and 2 deletions
Show stats Expand all files Collapse all files

2
pygrad/nn.py
View file

@ -48,7 +48,7 @@ def single_pass():
# Small test to see if autograd works. # Small test to see if autograd works.
def test(): def test():
# Input tensors. # Input tensors.
x, y, z = tensor.Tensor(-2), tensor.Tensor(5), tensor.Tensor(-4) x, y, z = tensor.Tensor(np.array([[1, 2, 3]])), tensor.Tensor(np.array([[2, 3, 4]])), tensor.Tensor(np.array([[1], [2], [3]]))
# Forward pass. # Forward pass.
q = x.add(y) q = x.add(y)

2
pygrad/tensor.py
View file

@ -27,7 +27,7 @@ class Tensor:
tensor._save(self, other) tensor._save(self, other)
def back(upstream): def back(upstream):
return np.dot(np.ones_like(self.value), upstream), np.dot(np.ones_like(self.value), upstream) return np.dot(np.ones_like(self.value).T, upstream), np.dot(np.ones_like(self.value).T, upstream)
tensor._back = back tensor._back = back
return tensor return tensor