one hot编码:`torch.Tensor.scatter_()`函数用法详解

torch.Tensor.scatter_()是torch.gather()函数的方向反向操作。两个函数可以看成一对兄弟函数。gather用来解码one hot,scatter_用来编码one hot。 scatter_(dim, index, src) → Tensor dim (python:int) – 用来寻址的坐标轴 index (LongTensor) – 索引 src(Tensor) –用来scatter的源张量,以防value未被指定。 value(python:float) – 用来scatter的源张量,以防src未被指定。 现在我们来看看具体这么用,看下面这个例子就一目了