tf.contrib.learn.multi_head
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. |