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

Update nn.test

Browse source
This commit is contained in:
Aodhnait Étaín 2020-11-15 19:45:05 +00:00
parent f85b9670b2
commit c1c48bb693
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/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)