pytorch 的标记默认从0开始tensor = torch.tensor([0, 2, 1, 3]) N = tensor.size(0) num_classes = 4 one_hot = torch.zeros(N, num_classes).long() ...
確定! 回上一頁