def linear_generator(x, output_dim, scope='Generator'): with tf.variable_scope(scope): return layers.linear(x, output_dim * output_dim)