def test_addmv(self): t1 = TensorBase(np.array([1, 2])) vec = TensorBase(np.array([1, 2, 3, 4])) mat = TensorBase(np.array([[2, 3, 3, 4], [5, 6, 6, 7]])) out = syft.addmv(t1, mat, vec, beta=2, alpha=2) self.assertTrue(np.array_equal(out.data, [68, 130]))
def addmv(self, mat, vec, beta=1, alpha=1): """"Performs a matrix-vector product of the matrix mat and the vector vec. The vector tensor is added to the final result. tensor1 and vec are 1d tensors out=(beta∗tensor)+(alpha∗(mat@vec2))""" return syft.addmv(self, mat, vec, beta, alpha)