提交 25652690 authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/opt_util.py

上级 00d5b577
from __future__ import absolute_import, print_function, division from __future__ import absolute_import, print_function, division
from functools import wraps from functools import wraps
import numpy import numpy as np
from theano import tensor, scalar as scal, Constant from theano import tensor, scalar as scal, Constant
from theano.gof import local_optimizer from theano.gof import local_optimizer
...@@ -11,7 +11,7 @@ from theano.tensor import (DimShuffle, get_scalar_constant_value, ...@@ -11,7 +11,7 @@ from theano.tensor import (DimShuffle, get_scalar_constant_value,
from .basic_ops import GpuFromHost, HostFromGpu, GpuAllocEmpty, GpuReshape, gpu_alloc_empty from .basic_ops import GpuFromHost, HostFromGpu, GpuAllocEmpty, GpuReshape, gpu_alloc_empty
from .elemwise import GpuDimShuffle, GpuElemwise from .elemwise import GpuDimShuffle, GpuElemwise
_one = scal.constant(numpy.asarray(1.0, dtype='float32')) _one = scal.constant(np.asarray(1.0, dtype='float32'))
def grab_cpu_scalar(v, nd): def grab_cpu_scalar(v, nd):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论