提交 6a2e8d50 authored 作者: Frederic's avatar Frederic

add the missing view_map to the Squeeze op.

上级 af64b4bb
......@@ -204,6 +204,7 @@ class SqueezeOp(theano.Op):
"""
def __init__(self, out_nd):
self.view_map = {0: [0]}
self.out_nd = out_nd
def __eq__(self, other):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论