IndexError：目标 60972032 超出范围

Question

我试图调用交叉熵损失，但它说索引超出范围

loss = nn.CrossEntropyLoss()
target = torch.empty(128, dtype=torch.long)
result = loss(output, target)

注意输出的形状为torch.Size([128, 10])

Answer 1

提供的示例中的target张量未初始化，请参阅torch.empty

它是空的，以修复该使用，例如.random_方法，如CrossEntropyLoss文档示例：

...
target = torch.empty(128, dtype=torch.long).random_(10)
...