import tensorflow as tf

                shape=[1, 1, n_basis, n_out],
                initializer=tf.random_normal_initializer())
            alpha_std = tf.exp(alpha_logstd)
            # Compute epsilon from {n_samples} standard Gaussian
