Python square_diagonalの例

コード例 #1

ファイル: test_sp.py プロジェクト: 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())

コード例 #2

ファイル: test_sp.py プロジェクト: 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())

コード例 #3

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

コード例 #4

ファイル: test_sp.py プロジェクト: huiwenhan/Theano

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