tf.contrib.learn.multi_head( heads, loss_weights=None )
Creates a MultiHead stemming from same logits/hidden layer. (deprecated)
THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Please switch to tf.contrib.estimator.*_head.
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.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.