Pytorch中Softmax、Log_Softmax、NLLLoss以及CrossEntropyLoss的关系与区别详解

最近看了一些Pytorch的代码,代码中使用了Log_Softmax方法,Loss函数使用了NLLLoss,作为深度学习新手,便上网查了一些资料,将相关知识总结记录以下。

本文主要参考了这篇文章,在此基础上加入了一些自己的理解。

Softmax

我们知道softmax激活函数的计算方式是对输入的每个元素值x求以自然常数e为底的指数,然后再分别除以他们的和,其计算公式如下:
S o f t m a x ( x i ) = e x p ( x i ) ∑ j e x p ( x j ) , . Softmax(x_i)=\frac{exp(x_i)}{\sum_{j}exp(x_j)},. Softmax(xi)=

F.log_softmax是一个函数,它在softmax的结果上再进行一次log运算。这个函数可以通过指定维度参数来计算log_softmax。在PyTorch,可以使用F.log_softmax(x, dim=1)或者F.log_softmax(x, dim=0)来调用log_softmax函数。 log_softmax函数的作用是将输入张量进行softmax操作后再进行log运算,得到一个新的张量。它常用于计算分类问题的概率分布。 具体地说,log_softmax函数可以将原始输入张量x的每个元素作为指数的指数,然后对结果进行归一化处理,最后取对数。这样可以得到一个概率分布,在该分布每个元素的取值范围为负无穷到0。 下面是一个使用log_softmax函数的示例代码: ```python import torch import torch.nn.functional as F inp1 = torch.Tensor([1, 3, 6]) print(inp1) print(F.log_softmax(inp1, dim=0)) ``` 在这个示例,inp1是一个输入张量,通过调用F.log_softmax(inp1, dim=0),我们可以得到对inp1进行log_softmax操作后的结果。 <span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [Pytorch学习笔记(七):F.softmax()和F.log_softmax函数详解](https://blog.csdn.net/m0_51004308/article/details/118001835)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* *3* [(详细全面)softmaxlog_softmax的联系和区别,NLLLOSS和CrossEntropyLoss的联系和区别](https://blog.csdn.net/qq_43391414/article/details/118421352)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值