nn.functional.softmax(X, dim=-1)

🕗 发布于 2024-10-13 09:12 深度学习 pytorch 机器学习

dim=-1表示在最后一个维度（大概率是一行）应用Softmax函数，将值标准化为概率分布。

实例

假设我们有一个张量X，形状为（2，3），内容如下：

import torch  
import torch.nn.functional as F  

X = torch.tensor([[1.0, 2.0, 3.0],  
                  [1.0, 2.0, 3.0]])  

# 计算 Softmax  
softmax_result = F.softmax(X, dim=-1)  
print(softmax_result)

输出：

tensor([[0.0900, 0.2447, 0.6652],  
        [0.0900, 0.2447, 0.6652]])

可以看到，每一行的输出值加起来为 1，这表示已经进行了Softmax操作。

原文地址：https://blog.csdn.net/weixin_46144186/article/details/142796744

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！

上一篇：2024年恩施职称评前公示
下一篇：基于卷积神经网络的书法字体识别系统，resnet50，mobilenet模型【pytorch框架+python】

mysql操作详解
当前页, 记录开始的位置(行数)计算。先定位到要查询哪个表,然后根据什么条件去查,表确定好了,条件也确定好了,开始利用select查询。select 聚合函数(列名) from 表名 group by
阅读更多2024-10-20
【C语言】深入理解指针（三）（下）
本篇文章将讲解以下知识：1、二维数组传参的本质2、函数指针变量3、函数指针数组。
阅读更多2024-10-20
Java Spring 中常用的 @PostConstruct 注解使用总结
通过本文，晓得嘞注解的适用场景、执行顺序和使用中的注意事项。我们通过两个实际案例展示了如何利用进行数据预热和配置文件加载。在项目中非常有用，特别是在需要依赖注入完成后自动执行某些初始化任务的场景。它不
阅读更多2024-10-20
elk部署安装
1、提前装好docker docker-compose相关命令2、替换docker仓库地址国内镜像源参考地址：https://blog.csdn.net/llc580231/article/detai
阅读更多2024-10-20
Gorm操作数据库，有和没有WithContext的区别
比如，如果用户请求已经取消，你可以随时停止相关的数据库操作，避免浪费资源。会直接执行操作，而不会受到外部上下文的控制。这意味着这些数据库操作在正常情况下会执行完毕，但你无法控制操作的超时、取消等行为。
阅读更多2024-10-20
hash冲突怎么解决？
然后以初始哈希值为起点，按照步长依次探测哈希表中的下一个位置，直到找到一个空位置为止。假设有数据项需要存储到哈希表中，首先使用哈希函数1计算出一个位置，如果该位置已被占用，就使用哈希函数2计算新位置，
阅读更多2024-10-20
windows的CMD命令提示符
输入help，查看帮助；Tab 键，自动补全；上/下方向键，查看历史命令；右键窗口标题栏 -> 属性，可以修改外观样式。
阅读更多2024-10-20
传感器黑电平箝位（Sensor black level clamping）
Sensor black level clamping是一种图像处理技术，用于固定图像传感器的黑电平。在图像传感器中，黑电平是指在没有光照条件下，传感器输出的信号电平。由于传感器本身存在暗电流和噪声，
阅读更多2024-10-20
9.存储过程安全性博客大纲（9/10）
存储过程（Stored Procedure）是一种在数据库管理系统中存储的程序，它由一组为了完成特定功能的SQL语句组成。这些SQL语句集合被编译并存储在数据库中，可以通过一个调用语句来执行。代码重用
阅读更多2024-10-20
代码训练营 day39|0-1背包问题，LeetCode 416
dp[i][j] 表示从下标为[0-i]的物品里任意取，放进容量为j的背包，价值总和最大是多少。(i 来表示物品、j表示背包容量)dp[j]表示：容量为j的背包，所背的物品价值可以最大为dp[j]今天
阅读更多2024-10-20

nn.functional.softmax(X, dim=-1)

实例

相关文章