Pytorch使用记录

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)