提交 17e9d105 authored 作者: Benjamin Scellier's avatar Benjamin Scellier

update comment

上级 84dead87
...@@ -456,10 +456,8 @@ class EighGrad(Op): ...@@ -456,10 +456,8 @@ class EighGrad(Op):
# implements the necessary logic. # implements the necessary logic.
out = self.tri0(g) + self.tri1(g).T out = self.tri0(g) + self.tri1(g).T
# The call to self.tri0 in perform upcast from float32 to # Make sure we return the right dtype even if NumPy performed
# float64 or from int* to int64 in numpy 1.6.1 but not in # upcasting in self.tri0.
# 1.6.2. We do not want version dependent dtype in Theano.
# We think it should be the same as the output.
outputs[0][0] = numpy.asarray(out, dtype=node.outputs[0].dtype) outputs[0][0] = numpy.asarray(out, dtype=node.outputs[0].dtype)
def infer_shape(self, node, shapes): def infer_shape(self, node, shapes):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论