提交 86e5bd8b authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Flake8 fixes.

上级 54bfc88a
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import copy import copy
from theano.compat import izip
import numpy import numpy
import theano import theano
...@@ -13,10 +12,9 @@ from theano.tensor.elemwise import (Elemwise, DimShuffle, CAReduceDtype) ...@@ -13,10 +12,9 @@ from theano.tensor.elemwise import (Elemwise, DimShuffle, CAReduceDtype)
try: try:
import pygpu import pygpu
from pygpu import gpuarray from pygpu import gpuarray
from pygpu.tools import ScalarArg, ArrayArg from pygpu.tools import ArrayArg
from pygpu.elemwise import ElemwiseKernel
from pygpu.reduction import ReductionKernel from pygpu.reduction import ReductionKernel
from pygpu.gpuarray import dtype_to_typecode, dtype_to_ctype from pygpu.gpuarray import dtype_to_typecode
except ImportError: except ImportError:
pass pass
...@@ -166,7 +164,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -166,7 +164,7 @@ class GpuElemwise(HideC, Elemwise):
'<gpuarray/elemwise.h>'] '<gpuarray/elemwise.h>']
def c_support_code_struct(self, node, name): def c_support_code_struct(self, node, name):
return "\nGpuElemwise *ge;\n"; return "\nGpuElemwise *ge;\n"
def c_init_code_struct(self, node, name, sub): def c_init_code_struct(self, node, name, sub):
inps, outs = self._get_vnames(node) inps, outs = self._get_vnames(node)
......
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
import os import os
import copy
import numpy import numpy
from six import integer_types from six import integer_types
from six.moves import StringIO from six.moves import StringIO
import theano
from theano import tensor, gof from theano import tensor, gof
from theano.tensor.subtensor import IncSubtensor, Subtensor, get_idx_list from theano.tensor.subtensor import IncSubtensor, Subtensor, get_idx_list
import theano.tensor.inplace
try: try:
import pygpu import pygpu
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论