什么是one-hot编码

one-hot编码即独热编码， 这是在多分类任务中经常使用的一种编码，对于softmax激活函数输出的各类别的概率， one-hot编码能够很好地用在计算损失上， 其方法是使用 N 位状态寄存器来对 N 个状态进行编码，每个状态都有它独立的寄存器位，并且在任意时候，其中只有一位有效

pytorch实现

nn.functional.one_hot()函数

输入：一组标签与类数量

输出：编码后的tensor