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

pep8 reflect

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