Pytorch 學習(8):Recurrent layers (循環層)實現之GRUCell

Pytorch 學習(8):Recurrent layers (循環層)實現之GRUCellapp GRU是Gated Recurrent Unit,GRU是LSTM的一個變化形式。先看一個GRUCell的小例子學習 rnn = nn.GRUCell(10, 20) input = Variable(torch.randn(6, 3, 10)) hx = Variable(torch.randn(
相關文章
相關標籤/搜索