提交 9a1ab3e8 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Add node comments preceding lines produced by aesara.link.utils.fgraph_to_python

上级 4ccf184f
...@@ -765,7 +765,7 @@ def fgraph_to_python( ...@@ -765,7 +765,7 @@ def fgraph_to_python(
node_output_names = [unique_name(v) for v in node.outputs] node_output_names = [unique_name(v) for v in node.outputs]
body_assigns.append( body_assigns.append(
f"{', '.join(node_output_names)} = {local_compiled_func_name}({', '.join(node_input_names)})" f"# {node}\n{', '.join(node_output_names)} = {local_compiled_func_name}({', '.join(node_input_names)})"
) )
fgraph_input_names = [unique_name(v) for v in fgraph.inputs] fgraph_input_names = [unique_name(v) for v in fgraph.inputs]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论