tensorflow中如何把一个batch图片种类logits如转化成列表的形式？

请问大家，比如我做一个狗狗的分类，分成3个文件夹
哈士奇标签为：0，吉娃娃标签：1，金毛标签为：2
并写入tfrecord文件了。当读出的是后每个样例的label标签就是，0或1或2代表是哪种狗。
每个batch比如是5则可能的一组标签值是[0,1，2,1,0],标示第一张图是哈士奇，第二个是吉娃娃，第三张是金毛。。

但是。。。。
我的神经网络最后是要有3个节点输出。比如:吉娃娃[1,0,0]; 哈士奇[0,1,0]; 金毛[0,0,1]
那我怎么把上面的这一个batch取值[0,1，2,1,0]，转换成[[1,0,0],[0,1,0],[0,0,1],[0,1,0],[1,0,0]]呢？
若label=[0,1，2,1,0]是一个tensor有没有办法，在label在sess.run（）取到实际值之前就用某个OP转换成想要的格式呢。。谢谢大家。