【发布时间】:2019-12-02 22:04:24
【问题描述】:
伙计们。 是否有 pytorch 等效于“tf.nn.rnn_cell.MultiRNNCell”可以堆叠多个单元格? 会不会是 torch.nn.LSTM()? 那么如何使用它来创建与 MultiRNNCell 相同的网络呢?
【问题讨论】:
标签: tensorflow pytorch lstm recurrent-neural-network
伙计们。 是否有 pytorch 等效于“tf.nn.rnn_cell.MultiRNNCell”可以堆叠多个单元格? 会不会是 torch.nn.LSTM()? 那么如何使用它来创建与 MultiRNNCell 相同的网络呢?
【问题讨论】:
标签: tensorflow pytorch lstm recurrent-neural-network
请参阅 num_layers 参数以获取 torch.nn.RNN。也适用于 LSTM 和 GRU。
【讨论】: