tensorflow 獲取bert的中間某一層輸出的方法

在微調bert做微調的時候，想着試一下中間層輸出進行分類，以及結合bert某幾層的輸出做分類。 一開始搜索到有博客說使用bert的pytorch版本可以非常容易獲取某一層的輸出，就想看一下源碼裏面最後一層 model.get_sequence_output()怎麼獲取的，發現了是先獲取bert每一層的輸出，然後取了最後一層。那問題就好弄了， 直接使用下面函數獲取所有層的輸出，然後取某一行進行後續操

>>阅读原文<<