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.
|