如何将输出张量转换为单热张量？

I need to calculate loss from the softmax output vs target. My target is like [0,0,1] and output is [0.3,0.3,0.4] For the purpose, prediction is correct. But a cost function of below type doesn't account for this kind of accuracyI need to calculate loss from the softmax outpu