Exemplos de Symbolic.compile_function_gradient em Python

Exemplo n.º 1

Arquivo: test_symbolic.py Projeto: adityachivu/projects

    def test_compile_function_gradient(self):
        def f(x):
            return x ** 2
        s = Symbolic(f)
        x = np.random.random((3, 4))
        o = s.trace(x)[1]
        self.assertRaises(TypeError, s.compile_function_gradient, x, o)

        F = s.compile_function_gradient(x, o.sum())
        sym_result = F(x)
        self.assertTrue(np.allclose(sym_result[0], f(x).sum()))
        self.assertTrue(np.allclose(sym_result[1], 2 * x))

Exemplo n.º 2

Arquivo: test_symbolic.py Projeto: southpawac/NeuroAIChallenge

    def test_compile_function_gradient(self):
        def f(x):
            return x ** 2
        s = Symbolic(f)
        x = np.random.random((3, 4))
        o = s.trace(x)[1]
        self.assertRaises(TypeError, s.compile_function_gradient, x, o)

        F = s.compile_function_gradient(x, o.sum())
        sym_result = F(x)
        self.assertTrue(np.allclose(sym_result[0], f(x).sum()))
        self.assertTrue(np.allclose(sym_result[1], 2 * x))