Python grad_scale示例

示例#1

文件： test_gradient.py 项目： michaelosthege/aesara

def test_grad_scale():
    x = theano.tensor.scalar()

    z = theano.tensor.grad(gradient.grad_scale(x, 2)**2, x)
    z2 = theano.tensor.grad(x**2, x)

    f = theano.function([x], outputs=[z, z2])

    if config.mode != "FAST_COMPILE":
        topo = f.maker.fgraph.toposort()
        assert not any(
            [isinstance(node.op, gradient.GradScale) for node in topo])
    out = f(2.0)

    assert np.allclose(out, (8, 4))

示例#2

文件： test_gradient.py 项目： Theano/Theano

def test_grad_scale():
    x = theano.tensor.scalar()

    z = theano.tensor.grad(gradient.grad_scale(x, 2)**2, x)
    z2 = theano.tensor.grad(x**2, x)

    f = theano.function([x], outputs=[z, z2])

    if theano.config.mode != "FAST_COMPILE":
        topo = f.maker.fgraph.toposort()
        assert not any([isinstance(node.op, gradient.GradScale)
                        for node in topo])
    out = f(2.)

    assert np.allclose(out, (8, 4))