Tag output is: [1, 0.5, 0.5, 0.5, 0.5.],
Where the maximum extraction index
labels_ = torch.max(labels, 1)[1]
### Back to index the maximum value
Tag output is: [1, 0.5, 0.5, 0.5, 0.5.],
Where the maximum extraction index
labels_ = torch.max(labels, 1)[1]
### Back to index the maximum value