tensorflow - 是否有与 Tensorflow 类“tf.nn.rnn_cell.MultiRNNCell”等效的 Pytorch？

Question

伙计们。是否有 pytorch 等效于“<a href="https://www.tensorflow.org/api_docs/python/tf/nn/rnn_cell/MultiRNNCell" rel="nofollow noreferrer">tf.nn.rnn_cell.MultiRNNCell”的堆栈多个细胞？会不会是 torch.nn.LSTM()？那么如何使用它来创建与 MultiRNNCell 相同的网络呢？

标签： tensorflowpytorchlstmrecurrent-neural-network

Answer 1

请参阅 的num_layers论点torch.nn.RNN。也适用于 LSTM 和 GRU。