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)
...