预期的标量类型 Long 但在 PyTorch 中使用 nn.CrossEntropyLoss() 找到 Float

Question

我想制作一个标签火炬张量。我选择了两种不同的方法，第一种方法在使用nn.CrossEntropyLoss() 计算损失的部分出错。我想知道为什么会这样，虽然张量结果是一样的。

第一种方法：

labels = torch.hstack((torch.zeros(100),torch.ones(100),1+torch.ones(100)))

第二种方法：

labels_np = np.vstack((np.zeros((100,1)),np.ones((100,1)),1+np.ones((100,1))))
labels = torch.squeeze(torch.tensor(labels_np).long())

错误：

expected scalar type Long but found Float in Pytoch

Answer 1

要修复您的第一种方法，您应该通过调用labels = labels.long() 来转换数据类型来更改类型：

labels = torch.hstack((torch.zeros(100),torch.ones(100),1+torch.ones(100)))
labels = labels.long()