batch_size = 32
thanks, I used It
target = torch.argmax(target, dim=1)
=>
tensor([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 1, 1], device=‘cuda:0’)
but It is disappear class information
[1, 0, 0, 0],  => class 1
[0, 1, 0, 0],  => class 2
[0, 0, 1, 0],  => class 3
[0, 0, 0, 1],  => class 4