提交 7611ad68 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3046 from harlouci/flake8

Flake8 for files in sandbox
"""Provides Ops for FFT and DCT. """Provides Ops for FFT and DCT.
""" """
from theano.gof import Op, Apply, generic
from theano import tensor
import numpy.fft
import numpy import numpy
import numpy.fft
from six.moves import xrange from six.moves import xrange
from theano import tensor
from theano.gof import Op, Apply, generic
class GradTodo(Op): class GradTodo(Op):
def make_node(self, x): def make_node(self, x):
return Apply(self, [x], [x.type()]) return Apply(self, [x], [x.type()])
def perform(self, node, inputs, outputs): def perform(self, node, inputs, outputs):
raise NotImplementedError('TODO') raise NotImplementedError('TODO')
grad_todo = GradTodo() grad_todo = GradTodo()
...@@ -45,8 +46,9 @@ class FFT(Op): ...@@ -45,8 +46,9 @@ class FFT(Op):
self.inverse = inverse self.inverse = inverse
def __eq__(self, other): def __eq__(self, other):
return type(self) == type(other) and (self.half == other.half) and (self.inverse == return (type(self) == type(other) and
other.inverse) self.half == other.half and
self.inverse == other.inverse)
def __hash__(self): def __hash__(self):
return hash(type(self)) ^ hash(self.half) ^ 9828743 ^ (self.inverse) return hash(type(self)) ^ hash(self.half) ^ 9828743 ^ (self.inverse)
...@@ -77,21 +79,22 @@ class FFT(Op): ...@@ -77,21 +79,22 @@ class FFT(Op):
else: else:
fft_fn = numpy.fft.fft fft_fn = numpy.fft.fft
fft = fft_fn(frames, int(n), int(axis)) fft = fft_fn(frames, int(n), int(axis))
if self.half: if self.half:
M, N = fft.shape M, N = fft.shape
if axis == 0: if axis == 0:
if (M % 2): if (M % 2):
raise ValueError('halfFFT on odd-length vectors is undefined') raise ValueError('halfFFT on odd-length vectors is undefined')
spectrogram[0] = fft[0:M/2, :] spectrogram[0] = fft[0:M / 2, :]
elif axis == 1: elif axis == 1:
if (N % 2): if (N % 2):
raise ValueError('halfFFT on odd-length vectors is undefined') raise ValueError('halfFFT on odd-length vectors is undefined')
spectrogram[0] = fft[:, 0:N/2] spectrogram[0] = fft[:, 0:N / 2]
else: else:
raise NotImplementedError() raise NotImplementedError()
else: else:
spectrogram[0] = fft spectrogram[0] = fft
def grad(self, inp, out): def grad(self, inp, out):
frames, n, axis = inp frames, n, axis = inp
g_spectrogram, g_buf = out g_spectrogram, g_buf = out
...@@ -112,9 +115,9 @@ def dct_matrix(rows, cols, unitary=True): ...@@ -112,9 +115,9 @@ def dct_matrix(rows, cols, unitary=True):
""" """
rval = numpy.zeros((rows, cols)) rval = numpy.zeros((rows, cols))
col_range = numpy.arange(cols) col_range = numpy.arange(cols)
scale = numpy.sqrt(2.0/cols) scale = numpy.sqrt(2.0 / cols)
for i in xrange(rows): for i in xrange(rows):
rval[i] = numpy.cos(i * (col_range*2+1)/(2.0 * cols) * numpy.pi) * scale rval[i] = numpy.cos(i * (col_range * 2 + 1) / (2.0 * cols) * numpy.pi) * scale
if unitary: if unitary:
rval[0] *= numpy.sqrt(0.5) rval[0] *= numpy.sqrt(0.5)
......
from __future__ import print_function from __future__ import print_function
import numpy, scipy.linalg
from theano import gof, tensor, scalar, function
import unittest import unittest
import numpy
from theano import gof, tensor, function
from theano.tests import unittest_tools as utt
class Minimal(gof.Op): class Minimal(gof.Op):
...@@ -49,7 +53,6 @@ minimal = Minimal() ...@@ -49,7 +53,6 @@ minimal = Minimal()
# TODO: test dtype conversion # TODO: test dtype conversion
# TODO: test that invalid types are rejected by make_node # TODO: test that invalid types are rejected by make_node
# TODO: test that each valid type for A and b works correctly # TODO: test that each valid type for A and b works correctly
from theano.tests import unittest_tools as utt
class T_minimal(unittest.TestCase): class T_minimal(unittest.TestCase):
......
...@@ -155,7 +155,6 @@ class MultinomialFromUniform(Op): ...@@ -155,7 +155,6 @@ class MultinomialFromUniform(Op):
unis_n = unis[n] unis_n = unis[n]
for m in range(nb_outcomes): for m in range(nb_outcomes):
z_nm = z[0][n, m]
cummul += pvals[n, m] cummul += pvals[n, m]
if (waiting and (cummul > unis_n)): if (waiting and (cummul > unis_n)):
z[0][n, m] = 1 z[0][n, m] = 1
...@@ -356,8 +355,8 @@ def local_gpu_multinomial(node): ...@@ -356,8 +355,8 @@ def local_gpu_multinomial(node):
return [host_from_gpu(gpu_op(*[gpu_from_host(i) return [host_from_gpu(gpu_op(*[gpu_from_host(i)
for i in node.inputs])).T] for i in node.inputs])).T]
if (isinstance(node.op, theano.sandbox.cuda.GpuFromHost) and if (isinstance(node.op, theano.sandbox.cuda.GpuFromHost) and
node.inputs[0].owner and type(node.inputs[0].owner.op) node.inputs[0].owner and
is MultinomialFromUniform): type(node.inputs[0].owner.op) is MultinomialFromUniform):
multi = node.inputs[0].owner multi = node.inputs[0].owner
p, u = multi.inputs p, u = multi.inputs
m, = multi.outputs m, = multi.outputs
......
...@@ -3,4 +3,6 @@ Neighbours was moved into theano.tensor.nnet.neighbours. ...@@ -3,4 +3,6 @@ Neighbours was moved into theano.tensor.nnet.neighbours.
This file was created for compatibility. This file was created for compatibility.
""" """
from theano.tensor.nnet.neighbours import (images2neibs, neibs2images, from theano.tensor.nnet.neighbours import (images2neibs, neibs2images,
Images2Neibs) Images2Neibs)
__all__ = ["images2neibs", "neibs2images", "Images2Neibs"]
from __future__ import print_function from __future__ import print_function
import numpy, scipy.linalg
from theano import gof, tensor, scalar
import unittest import unittest
import sys
import numpy
import scipy.linalg
import theano
from theano import gof, tensor, scalar
from theano.tests import unittest_tools as utt
class Solve(gof.Op): class Solve(gof.Op):
...@@ -32,7 +39,7 @@ class Solve(gof.Op): ...@@ -32,7 +39,7 @@ class Solve(gof.Op):
raise TypeError("b must be a matrix or vector", b_.type) raise TypeError("b must be a matrix or vector", b_.type)
odtype = scalar.upcast(A_.dtype, b_.dtype) odtype = scalar.upcast(A_.dtype, b_.dtype)
otype = tensor.TensorType(broadcastable=b_.broadcastable, dtype=odtype) otype = tensor.TensorType(broadcastable=b_.broadcastable, dtype=odtype)
return gof.Apply(op=self, inputs=[A, B], outputs=[otype()]) return gof.Apply(op=self, inputs=[A_, b_], outputs=[otype()])
def perform(self, node, inp, out): def perform(self, node, inp, out):
A, b = inp A, b = inp
...@@ -49,8 +56,6 @@ solve = Solve() ...@@ -49,8 +56,6 @@ solve = Solve()
# TODO: test dtype conversion # TODO: test dtype conversion
# TODO: test that invalid types are rejected by make_node # TODO: test that invalid types are rejected by make_node
# TODO: test that each valid type for A and b works correctly # TODO: test that each valid type for A and b works correctly
from theano.tests import unittest_tools as utt
class T_solve(unittest.TestCase): class T_solve(unittest.TestCase):
def setUp(self): def setUp(self):
......
...@@ -136,20 +136,14 @@ whitelist_flake8 = [ ...@@ -136,20 +136,14 @@ whitelist_flake8 = [
"sandbox/test_theano_object.py", "sandbox/test_theano_object.py",
"sandbox/test_scan.py", "sandbox/test_scan.py",
"sandbox/rng_mrg.py", "sandbox/rng_mrg.py",
"sandbox/solve.py",
"sandbox/theano_object.py", "sandbox/theano_object.py",
"sandbox/scan.py", "sandbox/scan.py",
"sandbox/multinomial.py",
"sandbox/neighbourhoods.py",
"sandbox/fourier.py",
"sandbox/test_multinomial.py", "sandbox/test_multinomial.py",
"sandbox/minimal.py",
"sandbox/test_rng_mrg.py", "sandbox/test_rng_mrg.py",
"sandbox/test_neighbourhoods.py", "sandbox/test_neighbourhoods.py",
"sandbox/symbolic_module.py", "sandbox/symbolic_module.py",
"sandbox/conv.py", "sandbox/conv.py",
"sandbox/debug.py", "sandbox/debug.py",
"sandbox/neighbours.py",
"sandbox/cuda/dnn.py", "sandbox/cuda/dnn.py",
"sandbox/cuda/var.py", "sandbox/cuda/var.py",
"sandbox/cuda/GpuConvGrad3D.py", "sandbox/cuda/GpuConvGrad3D.py",
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论