【pytorch】使用scatter_()进行one-hot编码