提交 121eeacc authored 作者: Saxenauts's avatar Saxenauts

Temporarily disables the warn_float64 flag in Arange Issue#

上级 4cd08584
...@@ -4957,7 +4957,7 @@ class ARange(Op): ...@@ -4957,7 +4957,7 @@ class ARange(Op):
inputs = [start, stop, step] inputs = [start, stop, step]
outputs = [tensor(self.dtype, (False,))] outputs = [tensor(self.dtype, (False,))]
return Apply(self, inputs, outputs) return Apply(self, inputs, outputs)
@theano.configparser.change_flags(warn_float64 = 'ignore')
def infer_shape(self, node, i_shapes): def infer_shape(self, node, i_shapes):
# Note start, stop and step can be float numbers. # Note start, stop and step can be float numbers.
start, stop, step = node.inputs start, stop, step = node.inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论