TensorRT IRNNv2Layer

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：