提交 2b9d83ae authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/blocksparse.py

上级 6fcac08d
...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division ...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division
import logging import logging
import os import os
import numpy import numpy as np
from theano import Apply, tensor from theano import Apply, tensor
from theano.gof import COp from theano.gof import COp
from theano.tensor import discrete_dtypes, as_tensor_variable from theano.tensor import discrete_dtypes, as_tensor_variable
...@@ -121,7 +121,7 @@ class GpuSparseBlockOuter(COp): ...@@ -121,7 +121,7 @@ class GpuSparseBlockOuter(COp):
def make_node(self, o, x, y, xIdx, yIdx, alpha=None): def make_node(self, o, x, y, xIdx, yIdx, alpha=None):
ctx = infer_context_name(o, x, y) ctx = infer_context_name(o, x, y)
one = tensor.constant(numpy.asarray(1.0, dtype='float32')) one = tensor.constant(np.asarray(1.0, dtype='float32'))
o = as_gpuarray_variable(o, ctx) o = as_gpuarray_variable(o, ctx)
x = as_gpuarray_variable(x, ctx) x = as_gpuarray_variable(x, ctx)
y = as_gpuarray_variable(y, ctx) y = as_gpuarray_variable(y, ctx)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论