提交 0706d9ad authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix local_join_make_vector docstring

上级 dc6d59c0
...@@ -2478,11 +2478,14 @@ def local_join_empty(fgraph, node): ...@@ -2478,11 +2478,14 @@ def local_join_empty(fgraph, node):
@register_useless @register_useless
@local_optimizer([Join]) @local_optimizer([Join])
def local_join_make_vector(fgraph, node): def local_join_make_vector(fgraph, node):
"""Join(0, make_vector1, make_vector2, ...) => Join(0, make_vector12, ...) r"""Merge `MakeVector` inputs within a `Join`.
Merge MakeVector inputs to Join. This can make the join completely For example:
disappear with the local_join_1 opt.
Join(0, make_vector1, make_vector2, ...) => Join(0, make_vector12, ...)
This in combination with the `local_join_1` optimization can make `Join`\s
completely disappear.
""" """
if not isinstance(node.op, Join) or node.outputs[0].ndim != 1: if not isinstance(node.op, Join) or node.outputs[0].ndim != 1:
return return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论