pytorch 损失函数

最新推荐文章于 2024-07-27 20:20:27 发布

圣托里尼的日落啊~

最新推荐文章于 2024-07-27 20:20:27 发布

阅读量843

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/qq_39504764/article/details/104946507

版权

pytorch 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

二分类交叉熵损失函数

import torch
import torch.nn.functional as F
pred = torch.tensor([[-0.4089,-1.2471,0.5907],[-0.4897,-0.8267,-0.7349],[0.5241,-0.1246,-0.4757]]).cuda()

print(pred)
'''
tensor([[-0.4089, -1.2471,  0.5907],
        [-0.4897, -0.8267, -0.7349],
        [ 0.5241, -0.1246, -0.4757]], device='cuda:0')
'''

v = torch.sigmoid(pred)
print(v)

'''
tensor([[0.3992, 0.2232, 0.6435],
        [0.3800, 0.3043, 0.3241],
        [0.6281, 0.4689, 0.3833]], device='cuda:0')
'''

target = torch.FloatTensor([[0,1,1],[0,0,1],[1,0,1]]).cuda()

print(F.binary_cross_entropy_with_logits(pred,target))
'''
tensor(0.7194, device='cuda:0')
'''

print(F.binary_cross_entropy(v,target))
'''
tensor(0.7194, device='cuda:0')
'''