提交 e9169843 authored 作者: hantek's avatar hantek

pep8 reflect

上级 8b62864b
...@@ -21,8 +21,6 @@ from .type import GpuArrayType, gpu_context_type ...@@ -21,8 +21,6 @@ from .type import GpuArrayType, gpu_context_type
from .basic_ops import (as_gpuarray_variable, HideC, GpuKernelBase, Kernel, from .basic_ops import (as_gpuarray_variable, HideC, GpuKernelBase, Kernel,
infer_context_name, gpu_contiguous) infer_context_name, gpu_contiguous)
import pdb
iadd_reg = {} iadd_reg = {}
......
...@@ -20,8 +20,6 @@ from ..type import gpuarray_shared_constructor ...@@ -20,8 +20,6 @@ from ..type import gpuarray_shared_constructor
from .config import mode_with_gpu from .config import mode_with_gpu
import pdb
class G_subtensor(test_subtensor.T_subtensor): class G_subtensor(test_subtensor.T_subtensor):
def shortDescription(self): def shortDescription(self):
...@@ -199,7 +197,7 @@ class test_gpuextractdiag(unittest.TestCase): ...@@ -199,7 +197,7 @@ class test_gpuextractdiag(unittest.TestCase):
GpuExtractDiag(offset, axis1, axis2)(x).eval({x: np_x}), GpuExtractDiag(offset, axis1, axis2)(x).eval({x: np_x}),
np_x.diagonal(offset, axis1, axis2)) np_x.diagonal(offset, axis1, axis2))
class test_gpuallocdiag(unittest.TestCase): class test_gpuallocdiag(unittest.TestCase):
def test_matrix(self): def test_matrix(self):
x = tensor.vector() x = tensor.vector()
...@@ -214,7 +212,7 @@ class test_gpuallocdiag(unittest.TestCase): ...@@ -214,7 +212,7 @@ class test_gpuallocdiag(unittest.TestCase):
def test_grad(self): def test_grad(self):
x = tensor.vector() x = tensor.vector()
np_x = np.random.randn(7).astype(theano.config.floatX) np_x = np.random.randn(7).astype(theano.config.floatX)
# offset = 0 case: # offset = 0 case:
mtx_x = GpuAllocDiag()(x) mtx_x = GpuAllocDiag()(x)
sum_mtx_x = tensor.sum(mtx_x) sum_mtx_x = tensor.sum(mtx_x)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论