2025年2月4日
pytorch | softmax(x,dim=-1)参数dim的理解
摘要： 转自：https://zhuanlan.zhihu.com/p/525276061 首先说一下Softmax函数，公式如下: 1. 三维tensor(C,H,W) 一般会设置成dim=0,1,2,-1的情况(可理解为维度索引)。其中2与-1等价，相同效果。 用一张图片来更好理解这个参数dim数值变化 阅读全文
posted @ 2025-02-04 16:28 有何m不可 阅读(53) 评论(0) 推荐(0) 编辑
点击右上角即可分享
微信分享提示