对pytorch中softmax(dim=x)的理解
先看看文档怎么说的
这里说指定某个纬度（dim)，就会让那个纬度的和为1
看一下input是2维的情况
dim=0
dim=1
再看看input是3维的情况
dim = 0
dim = 1
分类:
PyTorch
这里说指定某个纬度（dim)，就会让那个纬度的和为1
【推荐】国内首个AI IDE，深度理解中文开发场景，立即下载体验Trae
【推荐】编程新体验，更懂你的AI，立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包，你的智能百科全书，全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell：AI 加持，快人一步
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单，K8s 太复杂？w7panel 让容器管理更轻松！