gru_unit

    • gru_unit(input, hidden, size, param_attr=None, bias_attr=None, activation='tanh', gate_activation='sigmoid', origin_mode=False)[源代码]

    Gated Recurrent Unit（GRU）循环神经网络计算单元。该OP用于完成单个时间步内GRU的计算，支持以下两种计算方式：

    如果origin_mode为True，则使用的运算公式来自论文 。

    公式如下:

    gru_unit - 图2

    其中，

    • 参数:

    返回：Variable的三元组，包含三个与 input 相同数据类型的Tensor，分别表示下一时间步的隐状态（

    ）、重置的前一时间步的隐状态（ gru_unit - 图31 ）和 的拼接，形状分别为 gru_unit - 图33 、 和 gru_unit - 图35 。通常只有下一时间步的隐状态（ ）作为GRU的输出和隐状态使用，其他内容只是中间计算结果。

    返回类型: tuple