One -hot编码(一维数组、二维图像都可以):label = torch.nn.functional.one_hot(label, N)。 #一维数组的one hot编码,N为类别,label为数组.
確定! 回上一頁