import tensorflow as tf


    # Initialize the mean
    mean = tf.get_variable(name + "_mean", shape, dtype=dtype)

    # Initialize the standard deviation
    pre_sigma = tf.get_variable(name + "_standard_deviation",
                                         shape,
                                         initializer=std_init,
                                         dtype=dtype)

