提交 0abf9bd8 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix Images2Neibs bug.

上级 711f0835
......@@ -18,7 +18,7 @@ class Images2Neibs(Op):
def make_node(self, ten4, neib_shape):
ten4 = T.as_tensor_variable(ten4)
neib_shape = T.as_tensor_variable(neib_shape)
return Apply(self, [ten4, neib_shape], [T.matrix()])
return Apply(self, [ten4, neib_shape], [ten4.type()])
def grad(self, (pvals, unis), (gz,)):
return [None, None]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论