torch softmax log_softmax 以及dim的解析

綜合:將某維度的矩陣在某個維度的數據變成他們在權重上的比例分佈 softmax是一種特殊的歸一化 歸一化0-1的數字 而且都代表他們的佔比 概率分配 log就是在上面做了一次log運算 dim代表對1,2,3維矩陣進行歸一化計算的是按哪個維度進行計算 數字越大越深入 分別爲0,1,2 參考https://blog.csdn.net/sunyueqinghit/article/details/101
相關文章
相關標籤/搜索