Pytorch学习(二十九) ------------torch.split与torch.roll结合进行的一种高端操作
Howtoreverseamulti-hotencoding?multihot_batch=torch.tensor([[0,1,0,1],[0,0,0,1],[0,0,1,1]])(multihot_batch==torch.tensor(1)).nonzero()tensor([[0,1],[0,3],[1,3],[2,2],[2,3]])希望得到[[1,3],[3],[2,3]]其实像这种任