pytorch 三维one-hot tensor的制作