提交 9d6e9530 authored 作者: Frederic's avatar Frederic

python 2.4 fix for new Fourrier op.

上级 3fcec167
...@@ -89,7 +89,7 @@ class Fourier(gof.Op): ...@@ -89,7 +89,7 @@ class Fourier(gof.Op):
if len(shape_a) == 1: if len(shape_a) == 1:
return [(n,)] return [(n,)]
elif isinstance(axis, tensor.TensorConstant): elif isinstance(axis, tensor.TensorConstant):
out_shape = list(shape_a[0: axis.data]) + [n] +\ out_shape = list(shape_a[0: axis.data.item()]) + [n] +\
list(shape_a[axis.data + 1:]) list(shape_a[axis.data + 1:])
else: else:
l = len(shape_a) l = len(shape_a)
...@@ -105,7 +105,7 @@ class Fourier(gof.Op): ...@@ -105,7 +105,7 @@ class Fourier(gof.Op):
a = inputs[0] a = inputs[0]
n = inputs[1] n = inputs[1]
axis = inputs[2] axis = inputs[2]
output_storage[0][0] = numpy.fft.fft(a, n=int(n), axis=axis) output_storage[0][0] = numpy.fft.fft(a, n=int(n), axis=axis.item())
def grad(self, inputs, cost_grad): def grad(self, inputs, cost_grad):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论