logo
Loading...

Model summary參數值問題 - Cupoy

請教D69 HW model summary裡的lstm_1 layer的參數如何推得?

ml100-2,ml100-2-d69

Model summary參數值問題

2019/07/07 上午 09:47
機器學習共學討論版
鄧志偉
觀看數:110
回答數:1
收藏數:0
ml100-2
ml100-2-d69

請教D69 HW model summary裡的lstm_1 layer的參數如何推得?


回答列表

  • 2019/07/12 下午 00:44
    陳明佑 (Ming You Chen)
    贊同數:1
    不贊同數:0
    留言數:0

    同學您好 :

    有關LSTM的結構, 因為較為複雜 (屬於RNN的一種)

    有關結構與教學可能要另開一個 Part 解釋  

    因此這邊只針對您的問題 : "如何計算" 做答覆, 謝謝


    有關參數值的計算, 請對照參考這一篇連結

    https://zhuanlan.zhihu.com/p/39262960

    對照範例來說, 輸入的 x 有 512 組, 橫向傳遞的 h 維度 = c 維度 = 32

    每個 LSTM 結構中 W 的神經元都有 512+32 個外部傳進的連結 (X過來512, 前一個來32)

    因此 W 的權重個數總共是 (512+32)*32, 加上常數 b 有 32 個

    而這樣的 W 共有 4 組, 因此最後是

    4 * (32 * (512 + 32) + 32) = 69760