Ejemplo n.º 1
0
 def test_init_001(self):
     a = Tensor(np.array([1, 2, 3, 4, 5]), autograd=True)
     b = Tensor(np.array([10, 10, 10, 10, 10]), autograd=True)
     c = Tensor(np.array([5, 4, 3, 2, 1]), autograd=True)
     d = a + b
     e = b + c
     f = d + e
     print('f: {0};'.format(f.to_string()))
     print('d: {0};'.format(d.to_string()))
     print('e: {0};'.format(e.to_string()))
     print('a: {0};'.format(a.to_string()))
     print('b: {0};'.format(b.to_string()))
     print('c: {0};'.format(c.to_string()))
Ejemplo n.º 2
0
 def test_mul_backward_001(self):
     a = Tensor(np.array([1, 2, 3, 4, 5]), autograd=True)
     b = Tensor(np.array([10, 10, 10, 10, 10]), autograd=True)
     c = Tensor(np.array([5, 4, 3, 2, 1]), autograd=True)
     d = a + b
     e = b - c
     f = d * e
     f.backward(Tensor(np.array([1, 1, 1, 1, 1])))
     print('f: {0};'.format(f.to_string()))
     print('d: {0};'.format(d.to_string()))
     print('e: {0};'.format(e.to_string()))
     print('a: {0};'.format(a.to_string()))
     print('b: {0};'.format(b.to_string()))
     print('c: {0};'.format(c.to_string()))
Ejemplo n.º 3
0
 def test_sum_grad_001(self):
     v = Tensor(np.array([[1, 2, 3], [4, 5, 6]]), autograd=True)
     u = v.sum(0)
     u.backward(Tensor(np.array([1, 1, 1])))
     print('grad: {0};'.format(v.to_string()))