Pytorch学习笔记
CrossEntropyLoss示例
from torch import nn
import torch as t
sample=t.tensor([[2.,950],[1,500]])
target=t.tensor([1,1])
criterion = nn.CrossEntropyLoss()
loss=criterion(sample,target)
loss
输出
tensor(0.)
BCELoss示例
import torch as t
criterion=t.nn.BCELoss()
scores=t.tensor([1.,1.])
target=t.tensor([1.,1.])
loss=criterion(scores,target)
loss
输出
tensor(0.)
multilabel_soft_margin_loss示例
import torch.nn.functional as F
import torch as t
scores=t.tensor([[0.,20,0],[0,20,0]])
target=t.tensor([[0.,1.,0],[0.,1.,0]])
loss = F.multilabel_soft_margin_loss(scores, target)
loss
输出
tensor(0.5665)