ignore_index表示計(jì)算交叉熵時(shí)，自動(dòng)忽略的標(biāo)簽值，example：
import torch
import torch.nn.functional as F
pred = []
pred.append([0.9, 0.1])
pred.append([0.8, 0.2])
pred = torch.Tensor(pred).view(-1, 2)

label = torch.LongTensor([[1], [-1]]) # 這里輸出類別為0或1，-1表示不參與計(jì)算loss。且計(jì)算平均loss的時(shí)候，reduction只計(jì)算實(shí)際參與計(jì)算的個(gè)數(shù)，這里相當(dāng)于batchsize=2，但其中第index=1行為-1不參與計(jì)算loss。

#out = F.cross_entropy(pred.view(-1, 2), label.view(-1, ))
out = F.cross_entropy(pred.view(-1, 2), label.view(-1, ), ignore_index=-1)
print(out)

總結(jié)

以上是生活随笔為你收集整理的torch.nn.functional.cross_entropy.ignore_index的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。