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. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/learn/multi_head