import tensorflow as tf

        b1 = tf.get_variable('bias1', [1024], initializer=tf.constant_initializer(0.0))
        h1 = tf.nn.relu(tf.matmul(x, w1) + b1)
    with tf.variable_scope('layer2'):
        w2 = tf.get_variable('weight2', [1024, 1024], initializer=tf.random_normal_initializer())
        b2 = tf.get_variable('bias2', [1024], initializer=tf.constant_initializer(0.0))
        h2 = tf.nn.relu(tf.matmul(h1, w2) + b2)
    with tf.variable_scope('layer3'):
        w3 = tf.get_variable('weight3', [1024, 10], initializer=tf.random_normal_initializer())
