f7b10102a3a0b08272634347ff3a65efd3df18ce,test/functions/test_root_decomposition.py,TestRootDecomposition,test_root_decomposition,#TestRootDecomposition#,31

Before Change

        // Backward
        res.trace().backward()
        self.mat_clone.trace().backward()
        self.assertTrue(approx_equal(self.mat.grad, self.mat_clone.grad))

    def test_root_inv_decomposition(self):
        // Forward

After Change

        self.assertAllClose(res, mat)

        // Backward
        sum([mat.trace() for mat in res.view(-1, mat.size(-2), mat.size(-1))]).backward()
        sum([mat.trace() for mat in mat_clone.view(-1, mat.size(-2), mat.size(-1))]).backward()
        self.assertAllClose(mat.grad, mat_clone.grad)

Instances