交叉熵损失函数报错

Handsome_zhou

IndexError: Target 3 is out of bounds.

原因：交叉熵损失函数的输入矩阵元素大小不能超过2.

1. import torch
   
2. import torch.nn as nn
   
3. a = torch.tensor([[0.7,0.2,0.1]])
   
4. b = torch.tensor([2])
   
5. loss = nn.CrossEntropyLoss()
   
6. l = loss(a, b)
   
7. print(l)

复制代码