提交 8cf1e6cf authored 作者: abalkin's avatar abalkin

Fixed Eig.make_node().

上级 530453b1
...@@ -900,8 +900,8 @@ class Eig(Op): ...@@ -900,8 +900,8 @@ class Eig(Op):
def make_node(self, x): def make_node(self, x):
x = as_tensor_variable(x) x = as_tensor_variable(x)
w = theano.tensor.vector(x.dtype) w = theano.tensor.vector(dtype=x.dtype)
v = theano.tensor.matrix(x.dtype) v = theano.tensor.matrix(dtype=x.dtype)
return Apply(self, [x], [w, v]) return Apply(self, [x], [w, v])
def perform(self, node, (x,), (w, v)): def perform(self, node, (x,), (w, v)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论