如何將一個向量轉換爲相應的單向量向量？

Question

有一個形狀(batch_size,)的矢量。

它需要根據符號轉換爲形狀(batch_size, 2)。

基本上(1,-2, 1) -> ((1,0), (0,1), (1,0),)

是什麼在tensorflow做到這一點的好辦法？

Answer 1

您正在尋找tf.one_hot。