提交 1edf2dca authored 作者: Frederic's avatar Frederic

Make nonzero being printed correctly. Also get the hash and eq method correctly implemented.

上级 41b10b4a
......@@ -2051,6 +2051,8 @@ class Nonzero(gof.Op):
flattened input array.
"""
__props__ = ()
def make_node(self, a):
a = as_tensor_variable(a)
if a.ndim == 0:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论