pytorch lstm總結

文章目錄 1.batch_first作用 2.output[-1]和h_t區別 pytorch中lstm參數如下: 1.batch_first作用 我習慣把batch作爲輸入第一維, 所以就把batch_first設爲True. 結果跑實驗報錯, 提示維度有問題.調試後才發現batch_first只會影響input和output維度, h_t和c_t的維度仍然是[num_layers * num_
相關文章
相關標籤/搜索