test_sgld.py 文件源码

def test_psgld_dense(self):
        tf.reset_default_graph()

        x = tf.Variable(tf.zeros(20), dtype=tf.float32)
        loss = tf.reduce_sum(tf.square(x - 10))

        psgld = pSGLD(learning_rate=1.0)
        train_op_psgld = psgld.minimize(loss)

        sess = tf.InteractiveSession()
        sess.run(tf.global_variables_initializer())

        sess.run(train_op_psgld)
        xh = sess.run(x)