提交 fd698ba5 authored 作者: Mohammad Pezeshki's avatar Mohammad Pezeshki 提交者: Pascal Lamblin

better doc string

上级 dab5edfa
...@@ -4166,8 +4166,10 @@ register_canonicalize(local_reshape_chain(T.Reshape), ...@@ -4166,8 +4166,10 @@ register_canonicalize(local_reshape_chain(T.Reshape),
@gof.local_optimizer([T.Reshape]) @gof.local_optimizer([T.Reshape])
def local_useless_reshape(node): def local_useless_reshape(node):
""" """
Remove Reshape when both the input and the output have a Broadcastable dimensions in reshpae are replaced with dimshuffle.
single dimension. For example:
- reshape(x, (1, n)) --> dimshuffle{x,0}(reshape(x, (n,))
- reshape(x, (1, m, 1, n, 1, 1)) --> dimshuffle{x,0,x,1,x,x}(reshape(x, (m, n)))
""" """
op = node.op op = node.op
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论