Hello guys,
I’m probably just bad at searching. So pardon me if this is a repost.
I have a tensor with [batch_size, 4]
Andi Want the value of the 2nd dimension to be somehting like  [0,0,1,0], where the one corresponds to the max value in this tensor.
Any ideas?
thanks in advance!