tf.contrib.learn.multi_head

View source on GitHub

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

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.

A instance of Head that merges multiple heads.

ValueError if heads and loss_weights have different size.