1、法一
import torch
def one_hot(arr):
zero_arr=torch.zeros(len(arr),max(arr)+1)
zero_arr[torch.arange(len(arr)),arr]=1
return zero_arr
1、法一
import torch
def one_hot(arr):
zero_arr=torch.zeros(len(arr),max(arr)+1)
zero_arr[torch.arange(len(arr)),arr]=1
return zero_arr