提交 1a60236f authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Don't use ==, use T.eq.

上级 3b075b19
...@@ -454,7 +454,7 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None, ...@@ -454,7 +454,7 @@ def conv2d_fft(input, filters, image_shape=None, filter_shape=None,
raise ValueError('invalid mode') raise ValueError('invalid mode')
input_padded = T.opt.Assert("in conv2d_fft: width is not even")( input_padded = T.opt.Assert("in conv2d_fft: width is not even")(
input_padded, o1 % 2 == 0) input_padded, T.eq(o1 % 2, 0))
# reshape for FFT # reshape for FFT
input_flat = input_padded.reshape((b * ic, o0, o1)) input_flat = input_padded.reshape((b * ic, o0, o1))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论