tf.contrib.learn.multi_head
Создаёт MultiHead, исходя из одних и тех же logits/скрытых слоёв. (устаревшая функция)
tf.contrib.learn.multi_head(
heads, loss_weights=None
)
| Аргументы |
|---|
heads | список объектов Head. |
loss_weights | необязательный список весов, используемых для объединения потерь от каждого head. Все потери имеют одинаковый вес, если не указано иначе. |
| Возвращает |
|---|
Экземпляр Head, который объединяет несколько голов. |
| Исключения |
|---|
ValueError | если у heads и loss_weights разный размер. |
