pytorch中转one-hot的两种简单方法 1.替换零矩阵 arr = torch.tensor([2, 5, 6, 8, 6]) zero_arr = torch.zeros(arr.shape[0