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

Fix local_upcast_elemwise_constant_inputs docstring

上级 bd561974
...@@ -2281,10 +2281,7 @@ def local_upcast_elemwise_constant_inputs(fgraph, node): ...@@ -2281,10 +2281,7 @@ def local_upcast_elemwise_constant_inputs(fgraph, node):
@register_specialize @register_specialize
@local_optimizer([Rebroadcast]) @local_optimizer([Rebroadcast])
def local_useless_rebroadcast(fgraph, node): def local_useless_rebroadcast(fgraph, node):
""" """Remove `Rebroadcast` if it does not actually change the broadcasting pattern."""
Remove Rebroadcast if id does not actually change the broadcasting pattern.
"""
if isinstance(node.op, Rebroadcast): if isinstance(node.op, Rebroadcast):
x = node.inputs[0] x = node.inputs[0]
if np.all(x.broadcastable == node.outputs[0].broadcastable): if np.all(x.broadcastable == node.outputs[0].broadcastable):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论