print(one_hot) label=torch.topk(one_hot,1)[1].squeeze(1) print(label) 1 2 3 4 5 tensor([[0,0,1], [0,1,0], [0,1,0]]) tensor([2,1,1]) 以上這篇pytorch實現(xiàn)onehot編碼轉(zhuǎn)為普通label標(biāo)簽就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
www.dbjr.com.cn/article/1776...htm 2025-5-18