TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.contrib.learn.multi_head

View source on GitHub

Creates a MultiHead stemming from same logits/hidden layer. (deprecated)

tf.contrib.learn.multi_head(
    heads,
    loss_weights=None
)

Args:

  • heads: list of Head objects.
  • loss_weights: optional list of weights to be used to merge losses from each head. All losses are weighted equally if not provided.

Returns:

A instance of Head that merges multiple heads.

Raises:

  • ValueError: if heads and loss_weights have different size.