def testIsKronKron(self): # Tests _is_kron on a Kronecker matrix initializer = initializers.random_matrix(((2, 3), (3, 2)), tt_rank=1, dtype=self.dtype) kron_mat = variables.get_variable('kron_mat', initializer=initializer) self.assertTrue(kr._is_kron(kron_mat))
def testIsKronNonKron(self): # Tests _is_kron on a non-Kronecker matrix initializer = initializers.random_matrix(((2, 3), (3, 2)), tt_rank=2, dtype=self.dtype) tt_mat = variables.get_variable('tt_mat', initializer=initializer) self.assertFalse(kr._is_kron(tt_mat))
def testIsKronKron(self): # Tests _is_kron on a Kronecker matrix batch initializer = initializers.random_matrix_batch(((2, 3), (3, 2)), tt_rank=1, batch_size=3) kron_mat_batch = variables.get_variable('kron_mat_batch', initializer=initializer) self.assertTrue(kr._is_kron(kron_mat_batch))
def testIsKronNonKron(self): # Tests _is_kron on a non-Kronecker matrix batch initializer = initializers.random_matrix_batch(((2, 3), (3, 2)), tt_rank=2, batch_size=3) tt_mat_batch = variables.get_variable('tt_mat_batch', initializer=initializer) self.assertFalse(kr._is_kron(tt_mat_batch))