未验证 提交 da73bc39 编写于 作者: Z Zeng Jinle 提交者: GitHub

Merge pull request #14143 from sneaxiy/fix_dynamic_gru_bug

Fix dynamic_gru h_0 bug
...@@ -749,7 +749,7 @@ def dynamic_gru(input, ...@@ -749,7 +749,7 @@ def dynamic_gru(input,
attr=helper.bias_attr, shape=[1, 3 * size], dtype=dtype, is_bias=True) attr=helper.bias_attr, shape=[1, 3 * size], dtype=dtype, is_bias=True)
batch_size = input.shape[0] batch_size = input.shape[0]
inputs = {'Input': input, 'Weight': weight, 'Bias': bias} inputs = {'Input': input, 'Weight': weight, 'Bias': bias}
if h_0 != None: if h_0:
assert h_0.shape == ( assert h_0.shape == (
batch_size, size batch_size, size
), 'The shape of h0 should be(batch_size, %d)' % size ), 'The shape of h0 should be(batch_size, %d)' % size
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册