def testaddmm_2d(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]])) t1.addmm_(t2, mat, beta=2, alpha=2) self.assertTrue(np.array_equal(t1.data, [[10, 18], [12, 20]]))
def testaddmm_1d(self): t1 = TensorBase(np.array([1, 2, 3])) t2 = TensorBase(np.array([2, 3, 4])) mat = TensorBase(np.array([5])) t1.addmm_(t2, mat, beta=2, alpha=2) self.assertTrue(np.array_equal(t1.data, [50]))