Hi Simen!



 enemis:

tensor([[ 5.,  2.,  2.,  1.,  1.,  1.,  2.,  2.,  1.,  2.],
        [ 0., 13.,  2.,  3.,  1.,  0.,  0.,  0.,  0.,  0.]])



Please try torch.nn.functional.one_hot (x).sum (dim = 1).
Best.
K. Frank