Merge pull request #43 from lucasdelimanogueira/tmp

add example training

install.sh

@@ -0,0 +1,2 @@
+cd build
+make

norch/autograd/functions.py

@@ -27,7 +27,9 @@ class ElementwiseMulBackward:
         self.input = [x, y]
 
     def backward(self, gradient):
-        return [gradient * self.input[1], gradient * self.input[0]]
+        x = self.input[0]
+        y = self.input[1]
+        return [y * gradient, x * gradient]
     
 class MatmulBackward:
     def __init__(self, x, y):

norch/tensor.py

@@ -179,6 +179,8 @@ class Tensor:
         self.grad_fn = None
 
     def __getitem__(self, indices):
+        if isinstance(indices, int):
+            indices = [indices]
         if len(indices) != self.ndim:
             raise ValueError("Number of indices must match the number of dimensions")