pytorch實現onehot編碼轉為普通label標籤
阿新 • • 發佈：2020-01-09
label轉onehot的很多，但是onehot轉label的有點難找，所以就只能自己實現以下，用的topk函式，不知道有沒有更好的實現
one_hot = torch.tensor([[0,1],[0,1,0],0]]) print(one_hot) label = torch.topk(one_hot,1)[1].squeeze(1) print(label)
tensor([[0,0]]) tensor([2,1])
以上這篇pytorch實現onehot編碼轉為普通label標籤就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。