From c1c48bb693735a3a776d7d1fe70af842ac6966cb Mon Sep 17 00:00:00 2001 From: aodhneine Date: Sun, 15 Nov 2020 19:45:05 +0000 Subject: [PATCH] Update nn.test --- pygrad/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pygrad/nn.py b/pygrad/nn.py index d17a42a..a06366c 100644 --- a/pygrad/nn.py +++ b/pygrad/nn.py @@ -48,7 +48,7 @@ def single_pass(): # Small test to see if autograd works. def test(): # 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. q = x.add(y)