import tensorflow as tf

            self.b1 = tf.get_variable('b1', [1024],initializer=tf.constant_initializer(0.0))
            self.b2 = tf.get_variable('b2', [classnum],initializer=tf.constant_initializer(0.0))

    def inference(self,images):
        images=tf.cast(images,tf.float32)/255.0
        l1 = tf.matmul(images, self.w1)+self.b1
        l1=tf.nn.relu(l1)
        out = tf.matmul(l1, self.w2)+self.b2
        return out

