import tensorflow as tf

        if(h < len(layers_width)-2):
            # Perform affine mapping at each layer of the neural network
            Z = tf.layers.dense(Z, n_basis//2)
            # Define variational parameters
