使用一种热编码映射标签列表

Question

当'label'是列表时，我该怎么做？例如：label = [2,4,6,1,7 ...，9]

label = 3
NumClass = 10
NumRows  = 100

mask    =  torch.zeros(100, 64)
ones     =  torch.ones(1, 64)
ElementsPerClass = NumRows//NumClass
mask [ ElementsPerClass*label : ElementsPerClass*(label+1) ] = ones

Answer 1

您正在寻找scatter：

scatter