如何将一个向量转换为相应的单热向量？

Question

有一个形状为(batch_size,)的矢量。

需要根据符号将其转换为(batch_size, 2)形状。

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

在tensorflow中执行此操作的好方法是什么？

Answer 1

您正在寻找tf.one_hot。