Python trace示例

示例#1

文件： test_trace.py 项目： Beronx86/edward

def test_trace_scalar():
    X = tf.diag([2])
    with sess.as_default():
        assert trace(X).eval() == 2

示例#2

def test_trace_scalar():
    X = tf.diag([2])
    with sess.as_default():
        assert trace(X).eval() == 2

示例#3

文件： test_trace.py 项目： Beronx86/edward

def test_trace_mat():
    X = tf.diag(tf.ones([2]))
    with sess.as_default():
        assert trace(X).eval() == 2

示例#4

def test_trace_mat():
    X = tf.diag(tf.ones([2]))
    with sess.as_default():
        assert trace(X).eval() == 2