提交 31787931 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Remove Join(...) of only one tensor

上级 91f935b2
......@@ -1086,6 +1086,25 @@ def apply_rebroadcast_opt(rval):
changed = True
return rval
#############
# Join opts #
#############
@gof.local_optimizer([T.Join])
def local_join_1(node):
"""Join(i, x) => x
Remove Join() when only one element is joined.
"""
if not isinstance(node.op, T.Join):
return
axis = node.inputs[0]
tensors = node.inputs[1:]
if len(tensors) == 1:
return [tensors[0]]
##################
# Reshape opts #
##################
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论