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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]