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

Fix tensor.add to diff. matrices properly

Browse source
This commit is contained in:
Aodhnait Étaín 2020-11-15 19:45:51 +00:00
parent c1c48bb693
commit 52d266ef91
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
pygrad/tensor.py
View file

@ -27,7 +27,7 @@ class Tensor:
tensor._save(self, other)
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
return tensor