Ejemplos de square_diagonal en Python

Ejemplo n.º 1

Archivo: test_sp.py Proyecto: NicolasBouchard/Theano

def test_square_diagonal():
    for K in 1, 5:
        d = tensor.ivector()
        sd = sp.square_diagonal(d)
        f = theano.function([d], sd)
        n = numpy.zeros((K, K), dtype="int32")
        for i in range(K):
            n[i, i] = i

        assert numpy.all(n == f(range(K)).toarray())

Ejemplo n.º 2

Archivo: test_sp.py Proyecto: jsalvatier/Theano-1

def test_square_diagonal():
    for K in 1, 5:
        d = tensor.ivector()
        sd = sp.square_diagonal(d)
        f = theano.function([d], sd)
        n = numpy.zeros((K, K), dtype='int32')
        for i in range(K):
            n[i, i] = i

        assert numpy.all(n == f(range(K)).toarray())

Ejemplo n.º 3

 def d(x):
     return sp.sp_sum(sp.square_diagonal(x), sparse_grad=True)

Ejemplo n.º 4

Archivo: test_sp.py Proyecto: huiwenhan/Theano

 def d(x):
     return sp.sp_sum(sp.square_diagonal(x), sparse_grad=True)