pytorch 交叉损失熵函数及CLIP中的对比函数

交叉损失熵函数

在深度学习进行分类任务时经常用到交叉损失熵函数:首先定义logic表示余弦相似度,labels表示真实标签。可以直接使用交叉熵损失函数

import torch.nn as nn
Loss = nn.CrossEntropyLoss(logic, labels)

其中logic是一个NXC大小的数组,N表示有多少个样本(BatchSize),C表示类别总数。例如logic=\begin{bmatrix} 0.5\ 0.4\ 0.2\ 0.7 \\ 0.2\ 0.5\ 0.3\ 0.8 \\0.1\ 0.5\ 0.4\ 0.1\end{bmatrix}

        表示有三个样本,四个类别。

        而labels是一个一维数组,大小为N。表示N个样本所对应的真实类别。如label=[3,0,1],表示一个有三个样本,第一个样本对应的真实标签为3,第二个真实标签为0,第三个真实标签为1.

交叉熵函数计算公式为:

  • 0
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值