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

file theano/gpuarray/opt_util.py

上级 00d5b577
from __future__ import absolute_import, print_function, division
from functools import wraps
import numpy
import numpy as np
from theano import tensor, scalar as scal, Constant
from theano.gof import local_optimizer
......@@ -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 .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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论