def testaddcmul2d(self):
t1 = TensorBase(np.array([[1, 2], [1, 2]]))
t2 = TensorBase(np.array([[1, 2], [1, 2]]))
mat = TensorBase(np.array([[2, 3], [3, 4]]))
out = t1.addcmul(t2, mat, value=2)
self.assertTrue(np.array_equal(out.data, [[4, 11], [5, 12]]))
def testaddcmul1d(self):
t1 = TensorBase(np.array([1, 2, 3]))
t2 = TensorBase(np.array([2, 3, 4]))
mat = TensorBase(np.array([5]))
out = t1.addcmul(t2, mat, value=2)
self.assertTrue(np.array_equal(out.data, [9, 17, 29]))
