39f891bc57748795fb84d2bfefe359f53a02a5d2,test/functions/test_root_decomposition.py,TestRootDecompositionBatch,test_root_decomposition,#TestRootDecompositionBatch#,78

Before Change

    def test_root_decomposition(self):
        // Forward
        root = NonLazyTensor(self.mat).root_decomposition()
        res = root.matmul(root.transpose(-1, -2))
        self.assertTrue(approx_equal(res, self.mat))

        // Backward

After Change

    def test_root_decomposition(self):
        // Forward
        root = NonLazyTensor(self.mat).root_decomposition().root.evaluate()
        res = root.matmul(root.transpose(-1, -2))
        self.assertTrue(approx_equal(res, self.mat))

        // Backward

Instances