主要使用函数:torch.Tensor.scatter_(.....)
【sample】
将[1, 5, 4, 2]转换为one hot编码,形式如下:
[[0, 1, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 1],
[0, 0, 0, 0, 1, 0],
[0, 0, 1, 0, 0, 0]],其中每一行表示一个样本
函数调用:
torch.zeros(4, 6).scatter_(1, torch.tensor([1,5,4,2]).unsqueeze(1), 1)
说明...