元々やろうとしたことは、transformerの出力から各sentenceの末尾にあたるtokenのembeddingsだけを取り出すこと tensorflowは今までそこまで書いたことはなかったため、やってみると意外と時間がかかったためメモ 例として以下のような形を考える # 元のtensor (inputと呼ぶ) <tf.Tensor: id=88, shape=(2, 3, 4), dtype=int32, numpy= array([[[ 1, 2, 3, 4], [ 11, 22, 33, 44], [111, 222, 333, 444]], [[ 5, 6, 7, 8], [ 5…