tf.contrib.layers.legacy_linear

partial(func, *args, **keywords) - новая функция с частичным применением

tf.contrib.layers.legacy_linear(
    x, num_output_units, *, activation_fn=None, weight_init=._initializer at 0x7f58ea88c3b0>,
    bias_init=, name=None,
    weight_collections=('weights',), bias_collections=('biases',),
    output_collections=('activations',), trainable=True, weight_regularizer=None,
    bias_regularizer=None
)

указанных аргументов и ключевых слов.