在线观看不卡亚洲电影_亚洲妓女99综合网_91青青青亚洲娱乐在线观看_日韩无码高清综合久久

鍍金池/ 問(wèn)答/人工智能/ tensorflow中如何把一個(gè)batch圖片種類logits如[0,1,2,1

tensorflow中如何把一個(gè)batch圖片種類logits如[0,1,2,1,0]轉(zhuǎn)化成列表的形式?

請(qǐng)問(wèn)大家,比如我做一個(gè)狗狗的分類,分成3個(gè)文件夾
哈士奇標(biāo)簽為:0,吉娃娃標(biāo)簽:1,金毛標(biāo)簽為:2
并寫(xiě)入tfrecord文件了。當(dāng)讀出的是后每個(gè)樣例的label標(biāo)簽就是,0或1或2代表是哪種狗。
每個(gè)batch比如是5則可能的一組標(biāo)簽值是[0,1,2,1,0],標(biāo)示第一張圖是哈士奇,第二個(gè)是吉娃娃,第三張是金毛。。

但是。。。。
我的神經(jīng)網(wǎng)絡(luò)最后是要有3個(gè)節(jié)點(diǎn)輸出。比如:吉娃娃[1,0,0]; 哈士奇[0,1,0]; 金毛[0,0,1]
那我怎么把上面的這一個(gè)batch取值[0,1,2,1,0],轉(zhuǎn)換成[[1,0,0],[0,1,0],[0,0,1],[0,1,0],[1,0,0]]呢?
若label=[0,1,2,1,0]是一個(gè)tensor有沒(méi)有辦法,在label在sess.run()取到實(shí)際值之前就用某個(gè)OP轉(zhuǎn)換成想要的格式呢。。謝謝大家。

回答
編輯回答
乖乖噠

這種格式叫 one-hot

2017年9月13日 22:22