(TensorRT I RNNv2 Layer)

TensorRT IRNNv2Layer

IRNNv2Layer层用于实现递归结构，例如递归神经网络（RNN）、门控递归单元（GRU）以及长短期记忆（LSTM）。该层支持的类型包括RNN、GRU和LSTM。其执行一种递归运算，该运算由若干种知名的递归神经网络（RNN）“单元”中的一种所定义。

图层说明

该层接收输入序列X，初始隐藏状态H0，若所采用的单元为长短期记忆（LSTM）单元，则还需提供初始单元状态C0，并生成输出Y，该输出Y表示在T个时间步上计算得到的最终RNN“子层”的结果（详见下文）。此外，该层还可以选择性地生成输出hT，用以表示最终的隐藏状态；若所使用的单元为LSTM单元，则还会生成输出cT，用以表示最终的单元状态。

将单元的操作定义为函数G(x, h, c)。此函数接受向量形式的输入x、h和c，并产生最多两个向量形式的输出h’和c’，分别代表执行完单元操作后的隐藏状态与单元状态。

在默认配置（单向模式）下，RNNv2层按照如下图示的方式进行Gas操作：

![在这里插入图片描述](https://ad.itadn.com/c/weblog/blo