pytorch nn.LSTM()，nn.GRU()参数详解

LSTM结构中是一个神经网络，即上图的结构就是一个LSTM单元，里面的每一个黄框是一个神经网络，这个网络的隐藏单元个数咱们设为hidden_size，那么这个LSTM单元里就有4*hidden_size个参数。每一个LSTM输出的都是向量，包括 C t 和 h t C_t和h_t Ct​和ht​ ，它们的长度都是当前LSTM单元的hidden_sizehtml 函数 class torch.nn.

>>阅读原文<<