提交 05d008c7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Force MakeVector.perform to return an output of the expected type.

上级 98913c84
......@@ -250,7 +250,7 @@ class MakeVector(T.Op):
def __str__(self):
return self.__class__.__name__
def perform(self, node, inputs, (out,)):
out[0] = T.numpy.asarray(inputs)
out[0] = theano._asarray(inputs, dtype=node.outputs[0].dtype)
make_vector = MakeVector()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论