提交 519ac73c authored 作者: Reyhane Askari's avatar Reyhane Askari

_str_ method added to join

上级 0d7b4052
...@@ -65,7 +65,7 @@ array([8, 4]) ...@@ -65,7 +65,7 @@ array([8, 4])
>>> f = theano.function([x,y], z)# Do not take the shape. >>> f = theano.function([x,y], z)# Do not take the shape.
>>> theano.printing.debugprint(f) # doctest: +NORMALIZE_WHITESPACE >>> theano.printing.debugprint(f) # doctest: +NORMALIZE_WHITESPACE
Join{view=-1} [id A] '' 0 Join [id A] '' 0
|TensorConstant{0} [id B] |TensorConstant{0} [id B]
|x [id C] |x [id C]
|y [id D] |y [id D]
......
...@@ -3895,6 +3895,12 @@ class Join(Op): ...@@ -3895,6 +3895,12 @@ class Join(Op):
# start from index 1. # start from index 1.
self.view_map = {0: [1 + view]} self.view_map = {0: [1 + view]}
def __str__(self):
if self.view == -1:
return "Join"
else:
return super(Join, self).__str__()
def __setstate__(self, d): def __setstate__(self, d):
self.__dict__.update(d) self.__dict__.update(d)
if not hasattr(self, "view"): if not hasattr(self, "view"):
...@@ -4070,7 +4076,7 @@ class Join(Op): ...@@ -4070,7 +4076,7 @@ class Join(Op):
} }
Py_XDECREF(%(out)s); Py_XDECREF(%(out)s);
%(out)s = (PyArrayObject *)PyArray_Concatenate(list, axis); %(out)s = (PyArrayObject *)PyArray_Concatenate(list, axis);
(list); Py_DECREF(list);
if(!%(out)s){ if(!%(out)s){
%(fail)s %(fail)s
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论