GRUCell
- class
GRUCell
(hidden_size, param_attr=None, bias_attr=None, gate_activation=None, activation=None, dtype="float32", name="GRUCell")[源代码]
公式如下:
更多细节可以参考
- 参数:
- hidden_size (int) - GRUCell中的隐藏层大小。
- bias_attr (ParamAttr,可选) - 指定偏置参数属性的对象。默认值为None,表示使用默认的偏置参数属性。具体用法请参见 ParamAttr 。
- gate_activation (function,可选) -
的激活函数。 默认值为
fluid.layers.sigmoid
。 - activation (function,可选) - 的激活函数。 默认值为
- dtype (string,可选) - 此cell中使用的数据类型。 默认为"float32"。
- name (string,可选) - 用于标识参数和偏差的名称域。
返回:GRUCell类的实例对象。
执行GRU的计算。
- 参数:
- input (Variable) - 输入,形状为
的tensor,对应于公式中的 。数据类型应为float32。
- states (Variable) - 状态,形状为
的tensor。 对应于公式中的 。数据类型应为float32。
- input (Variable) - 输入,形状为
返回:一个元组 (outputs, new_states)
,其中 outputs
和 是同一个tensor,其形状为
,数据类型和
state
的数据类型相同,对应于公式中的
。
返回类型:tuple
(batch大小为-1,自动插入到形状中),对应于
的形状。
参数:无。
返回:GRUCell的 。
返回类型:Variable