提交 c0f6f492 编写于 作者: Y yangyaming

Add shape info for arguments.

上级 d6ec9630
...@@ -1199,9 +1199,9 @@ def lstm_unit(x_t, ...@@ -1199,9 +1199,9 @@ def lstm_unit(x_t,
This layer has two outputs including :math:`h_t` and :math:`o_t`. This layer has two outputs including :math:`h_t` and :math:`o_t`.
Args: Args:
x_t (Variable): The input value of current step. x_t (Variable): The input value of current step, a 2-D tensor.
hidden_t_prev (Variable): The hidden value of lstm unit. hidden_t_prev (Variable): The hidden value of lstm unit, a 2-D tensor.
cell_t_prev (Variable): The cell value of lstm unit. cell_t_prev (Variable): The cell value of lstm unit, a 2-D tensor.
forget_bias (float): The forget bias of lstm unit. forget_bias (float): The forget bias of lstm unit.
param_attr (ParamAttr): The attributes of parameter weights, used to set param_attr (ParamAttr): The attributes of parameter weights, used to set
initializer, name etc. initializer, name etc.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册