human
I wrote the below code

    def build():
        'Builds    loss    graph.'
        x = tf.get_variable('x', shape=[batch_size, num_dims], dtype=dtype, initializer=tf.random_normal_initializer(stddev=stddev))
        w = tf.get_variable('w', shape=[batch_size, num_dims, num_dims], dtype=dtype, initializer=tf.random_uniform_initializer(), trainable=False)
        y = tf.get_variable('y', shape=[batch_size, num_dims], dtype=dtype, initializer=tf.random_uniform_initializer(), trainable=False)
        product = tf.squeeze(tf.matmul(w, tf.expand_dims(x, (-1))))
        return tf.reduce_mean(tf.reduce_sum(((product - y) ** 2), 1))
    return build

What's a good function header?

Here's a good function header?
assistant
#Here's a good function header.

def quadratic(batch_size=128, num_dims=10, stddev=0.01, dtype=tf.float32):