torch.LongTensor轉換成one hot tensor編碼

import torch ''' pytorch0.4.0及以上的高版本，加上pytorch0.2.0都支持torch.Tensor._scatter 功能實現將torch.LongTensor編碼成 one-hot vector的功能 ''' rois_label=torch.tensor( [[0,1,0,5,2,7,4,3,1,2]] )#假設當前的ground truth boxe

>>阅读原文<<