pytorch的softmax

《pytorch的softmax》
《pytorch的softmax》

softmax有如上的用法

当dim=0,则第一行的值为 《pytorch的softmax》 ,以第0维为目标

剩下的以此类推

之前没有遇到过,因此在这里记下笔记

    原文作者:李元芳
    原文地址: https://zhuanlan.zhihu.com/p/49917021
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞