提交 140c9ffb authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Merged

import theano
from theano.tensor import basic as T
import numpy as N
from theano.sandbox.cuda import cuda_available, cuda_enabled
#from util import strutil
from theano import printing
from theano.tensor.blas_headers import blas_header_text
......@@ -570,6 +569,6 @@ def computeH(V,W,b,d):
from . import ConvGrad3D
from . import ConvTransp3D
if cuda_available:
from . import GpuConv3D
#from theano.sandbox.cuda import cuda_available, cuda_enabled
#if cuda_available:
# from . import GpuConv3D
import theano
from theano.tensor import basic as T
from theano.sandbox.cuda import cuda_available, cuda_enabled
#from util import strutil
import numpy as N
......@@ -269,6 +268,7 @@ convGrad3D = ConvGrad3D()
from Conv3D import conv3D
from ConvTransp3D import convTransp3D
if cuda_available:
from theano.sandbox.cuda import CudaNdarrayType, float32_shared_constructor
from . import GpuConvGrad3D
#from theano.sandbox.cuda import cuda_available, cuda_enabled
#if cuda_available:
# from theano.sandbox.cuda import CudaNdarrayType, float32_shared_constructor
# from . import GpuConvGrad3D
......@@ -2,7 +2,6 @@ import numpy as N
from theano.tensor import basic as T
#from util import strutil
import theano
from theano.sandbox.cuda import cuda_available, cuda_enabled
class ConvTransp3D(theano.Op):
""" "Transpose" of Conv3D (Conv3D implements multiplication by an implicitly defined matrix W. This implements multiplication by its transpose) """
......@@ -405,5 +404,6 @@ def computeR(W,b,d,H,Rshape = None):
from Conv3D import conv3D
from ConvGrad3D import convGrad3D
if cuda_available:
import GpuConvTransp3D
#from theano.sandbox.cuda import cuda_available, cuda_enabled
#if cuda_available:
# import GpuConvTransp3D
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论