提交 a7849db5 编写于 作者: M minqiyang

Port new added code

上级 6c775a0a
...@@ -37,7 +37,7 @@ def attention_lstm( ...@@ -37,7 +37,7 @@ def attention_lstm(
T = sum(lod[0]) T = sum(lod[0])
N = len(lod[0]) N = len(lod[0])
M = x.shape[1] M = x.shape[1]
D = b.shape[1] / 4 D = b.shape[1] // 4
assert T == x.shape[0] assert T == x.shape[0]
assert len(fcws) == len(fcbs) assert len(fcws) == len(fcbs)
hidden = [] hidden = []
......
...@@ -159,7 +159,7 @@ def program_to_code(prog): ...@@ -159,7 +159,7 @@ def program_to_code(prog):
get_indent_space(indent), '{', block_idx)) get_indent_space(indent), '{', block_idx))
indent += 1 indent += 1
# sort all vars # sort all vars
all_vars = sorted(block.vars.iteritems(), key=lambda x: x[0]) all_vars = sorted(six.iteritems(block.vars), key=lambda x: x[0])
for var in all_vars: for var in all_vars:
print("{}{}".format( print("{}{}".format(
get_indent_space(indent), variable_to_code(var[1]))) get_indent_space(indent), variable_to_code(var[1])))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册